Update app.py

app.py

@@ -53,12 +53,18 @@ def scrape_lora_link(url):
 
         # Use regular expression to find the link
         pattern = r'href="(.*?lora.*?\.safetensors\?download=true)"'
+        pattern2 = r'href="(.*?\.safetensors\?download=true)"'
         match = re.search(pattern, content)
+        match2 = re.search(pattern2, content)
 
         if match:
-           safetensors_url = match.group(1)
-           filename = safetensors_url.split('/')[-1].split('?')[0]  # Extract the filename from the URL
-           return filename
+            safetensors_url = match.group(1)
+            filename = safetensors_url.split('/')[-1].split('?')[0]  # Extract the filename from the URL
+            return filename
+        elif match2:
+            safetensors_url = match2.group(1)
+            filename = safetensors_url.split('/')[-1].split('?')[0]
+            return filename
         else:
             return None
     except requests.RequestException as e:
@@ -80,6 +86,7 @@ def enable_lora(lora_add,progress=gr.Progress(track_tqdm=True)):
         else:
             try:
                 pipe.load_lora_weights(lora_add)
+                print(f'lora loading: {lora_add}')
                 gr.Info(f"{lora_add} Loaded")
                 return gr.update(label="LoRA Loaded Now")
             except: