Update model_download.py

IAHispano · Jan 29, 2024 · 7a14fda · 7a14fda
1 parent 7350e69
commit 7a14fda
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/rvc/lib/tools/model_download.py b/rvc/lib/tools/model_download.py
@@ -195,6 +195,10 @@ def unzip_file(zip_path, zip_file_name):
 
 
 url = sys.argv[1]
+
+if "?download=true" in url:
+    url = url.replace("?download=true", "")
+
 verify = download_from_url(url)
 
 if verify == "downloaded":
@@ -217,7 +221,7 @@ def unzip_file(zip_path, zip_file_name):
                 print(f"Error downloading {model_name}")
                 sys.exit()
     if extract_folder_path == "":
-        print("No zip founded...")
+        print("Zip file was not found.")
         sys.exit()
     result = search_pth_index(extract_folder_path)
 else: