Update app.py

app.py

@@ -53,15 +53,14 @@ def scrape_lora_link(url):
            return filename
         else:
             return None
-
     except requests.RequestException as e:
         raise gr.Error(f"An error occurred while fetching the URL: {e}")
-        return None
 
 def enable_lora(lora_scale,lora_add):
     if not lora_add:
         gr.Info("No Lora Loaded, Use basemodel")
     else:
+        pipe.unload_lora_weights()
         url = f'https://huggingface.co/{lora_add}/tree/main'
         lora_name = scrape_lora_link(url)
         if lora_name:
@@ -69,7 +68,11 @@ def enable_lora(lora_scale,lora_add):
             pipe.fuse_lora(lora_scale=lora_scale)
             gr.Info(f"{lora_add} Loaded")
         else:
-            gr.Warning(f"{lora_add} Load fail, check again.")
+            try:
+                pipe.load_lora_weights(lora_add)
+                pipe.fuse_lora(lora_scale=lora_scale)
+            except:
+                raise gr.Error(f"{lora_add} Load fail, check again.")
 
 @spaces.GPU()
 def generate_image(
@@ -223,6 +226,4 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
         api_name="run",
     )
     
-
-    
 demo.queue().launch()