Update infer/lib/train/process_ckpt.py

infer/lib/train/process_ckpt.py

@@ -54,7 +54,7 @@ from google.oauth2 import service_account
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaFileUpload
 
-def savee(ckpt, sr, if_f0, name, epoch, version, hps, credentials_path):
+def savee(ckpt, sr, if_f0, name, epoch, version, hps):
     try:
         opt = {}
         opt["weight"] = {}
@@ -91,6 +91,7 @@ def savee(ckpt, sr, if_f0, name, epoch, version, hps, credentials_path):
         checkpoint_file = "assets/weights/%s.pth" % name
         torch.save(opt, checkpoint_file)
 
+        credentials_path = "/credentials.json"
         # Set up Google Drive API and upload the file
         SCOPES = ['https://www.googleapis.com/auth/drive.file']
         creds = service_account.Credentials.from_service_account_file(credentials_path, scopes=SCOPES)
@@ -98,7 +99,7 @@ def savee(ckpt, sr, if_f0, name, epoch, version, hps, credentials_path):
         drive_service = build('drive', 'v3', credentials=creds)
 
         # Specify the file name for Google Drive
-        drive_file_name = 'your_drive_file_name_here'
+        drive_file_name = name + ".pth"
 
         # Create a MediaFileUpload object to upload the file
         media = MediaFileUpload(checkpoint_file, resumable=True)