Update models/model.py

models/model.py

@@ -5,6 +5,8 @@ import requests
 import json
 import torch
 import os
+import re
+import urllib.request
 from inference import Inference
 import gradio as gr
 from constants import VOICE_METHODS, BARK_VOICES, EDGE_VOICES, zips_folder, unzips_folder
@@ -66,7 +68,18 @@ def compress(modelname, files):
     return file_path
 
 # download the model only
+def get_username(url):
+    match_username = re.search(r'models/(.*?)/', url)
+    if match_username:
+        result = match_username.group(1)
+        return result
 
+def get_username_hf(url):
+    match_username = re.search(r'huggingface.co/(.*?)/', url)
+    if match_username:
+        result = match_username.group(1)
+        return result
+        
 def download_online_model(url):
     url = url.strip()
     if url.startswith('https://download.openxlab.org.cn/models/'):