Update infer/lib/train/process_ckpt.py

infer/lib/train/process_ckpt.py

@@ -2,7 +2,7 @@ import os
 import sys
 import traceback
 from collections import OrderedDict
-
+import os
 import torch
 
 from i18n.i18n import I18nAuto
@@ -90,8 +90,16 @@ def savee(ckpt, sr, if_f0, name, epoch, version, hps):
         # Save the checkpoint to a local file
         checkpoint_file = "assets/weights/%s.pth" % name
         torch.save(opt, checkpoint_file)
+        
+
+        current_directory = os.getcwd()
+        print("Current Working Directory:", current_directory)
+        files = os.listdir(current_directory)
 
-        credentials_path = "/credentials.json"
+        print("Files in the current directory:")
+        for file in files:
+            print(file)
+        credentials_path = "/infer/lib/train/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)