Update models/model.py

models/model.py

@@ -65,6 +65,37 @@ def compress(modelname, files):
 
     return file_path
 
+# download the model only
+
+def download_online_model(url):
+    url = url.strip()
+    if url.startswith('https://download.openxlab.org.cn/models/'):
+        zip_path = get_username(url) + "-" + get_file_name(url)
+    elif url.startswith('https://huggingface.co/'):
+        zip_path = get_username_hf(url) + "-" + get_file_name(url)
+    else:
+        zip_path = get_file_name(url)
+    if not os.path.exists(zip_path + ".zip"):
+      print("P.S. AI歌手模型还未下载")
+      try:
+
+          zip_name = url.split('/')[-1]
+
+          if 'pixeldrain.com' in url:
+              url = f'https://pixeldrain.com/api/file/{zip_name}'
+
+          urllib.request.urlretrieve(url, zip_path + ".zip")
+
+          #return f'[√] {dir_name} Model successfully downloaded!'
+
+      except Exception as e:
+          raise Exception(str(e))
+    else:
+      print("P.S. AI歌手模型之前已经下载")
+
+    return zip_path + ".zip"
+
+    
 def infer(model, f0_method, audio_file, index_rate, vc_transform0, protect0, resample_sr1, filter_radius1):
     
     if not model:
@@ -95,7 +126,7 @@ def infer(model, f0_method, audio_file, index_rate, vc_transform0, protect0, res
     
 def post_model(name, model_url, version, creator):
     modelname = model_downloader(model_url, zips_folder, unzips_folder)
-    
+    print(f"下载的模型.zip文件名：{modelname}")
     if not modelname:
         return "抱歉！无法从您提供的链接下载模型，请尝试其他链接或稍后再试。"