minor change; moved ensure package inside when needed

2024-05-03 13:01:42 +05:30
parent ebfd510ae3
commit 9d15ec458e
1 changed files with 2 additions and 1 deletions
@@ -101,7 +101,8 @@ def download_model_restore(model_restore_path):
        print("Model file does not exist, downloading...")
        
        # Google Drive ID for the file
-        file_id = '1ruJg4lqR_jgQPj-9K0PP-L2vJERYOxLP'
+        # file_id = '1ruJg4lqR_jgQPj-9K0PP-L2vJERYOxLP'
+        file_id="1AVVLm1LxOs3W1Fp_GLefIz6fdWfEYg88"
        gdown.download(id=file_id, output=model_restore_path, quiet=False)
        print("Download complete.")
    else: