Update infer/lib/train/process_ckpt.py

infer/lib/train/process_ckpt.py

@@ -9,9 +9,13 @@ 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:
+        print("ckpt, sr, if_f0, name, epoch, version, hps", ckpt, sr, if_f0, name, epoch, version, hps)
         opt = OrderedDict()
         opt["weight"] = {}
         for key in ckpt.keys():
@@ -43,6 +47,20 @@ 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()