Update app.py

app.py

@@ -829,10 +829,15 @@ def save_wav(file):
             return None
 
 def download_from_url(url, model):
+    if model =='':
+        try:
+            model = url.split('/')[-1]
+        except:
+            return "You need to name your model. For example: My-Model"
+    url=url.replace('/blob/main/','/resolve/main/')
+    model=model.replace('.pth','').replace('.index','').replace('.zip','')
     if url == '':
         return "URL cannot be left empty."
-    if model =='':
-        return "You need to name your model. For example: My-Model"
     url = url.strip()
     zip_dirs = ["zips", "unzips"]
     for directory in zip_dirs:
@@ -843,6 +848,13 @@ def download_from_url(url, model):
     zipfile = model + '.zip'
     zipfile_path = './zips/' + zipfile
     try:
+        if url.endswith('.pth'):
+            subprocess.run(["wget", url, "-O", f'./assets/weights/{model}.pth'])
+            return f"Sucessfully downloaded as {model}.pth"
+        if url.endswith('.index'):
+            if not os.path.exists(f'./logs/{model}'): os.makedirs(f'./logs/{model}')
+            subprocess.run(["wget", url, "-O", f'./logs/{model}/added_{model}.index'])
+            return f"Successfully downloaded as added_{model}.index"
         if "drive.google.com" in url:
             subprocess.run(["gdown", url, "--fuzzy", "-O", zipfile_path])
         elif "mega.nz" in url: