Update infer/lib/train/process_ckpt.py

infer/lib/train/process_ckpt.py

@@ -9,9 +9,6 @@ from i18n.i18n import I18nAuto
 
 i18n = I18nAuto()
 
-import os
-from pydrive.auth import GoogleAuth
-from pydrive.drive import GoogleDrive
 
 def savee(ckpt, sr, if_f0, name, epoch, version, hps):
     try:
@@ -47,20 +44,7 @@ def savee(ckpt, sr, if_f0, name, epoch, version, hps):
         opt["f0"] = if_f0
         opt["version"] = version
         torch.save(opt, "assets/weights/%s.pth" % name)
-        
-        local_file_path = "assets/weights/%s.pth" % name
-        torch.save(opt, local_file_path)
-
-        # Authenticate with Google Drive
-        gauth = GoogleAuth()
-        gauth.LocalWebserverAuth()  # This will open a web page for authentication
-        drive = GoogleDrive(gauth)
-
-        # Create a Google Drive file and upload it
-        file_drive = drive.CreateFile({'title': name})
-        file_drive.Upload()
 
-        return "Success: File uploaded to Google Drive."
         return "Success."
     except:
         return traceback.format_exc()