Update app.py

app.py

@@ -55,7 +55,7 @@ def scrape_lora_link(url):
             return None
 
     except requests.RequestException as e:
-        print(f"An error occurred while fetching the URL: {e}")
+        raise gr.Error(f"An error occurred while fetching the URL: {e}")
         return None
 
 def enable_lora(lora_scale,lora_add):
@@ -63,10 +63,13 @@ def enable_lora(lora_scale,lora_add):
         gr.Info("No Lora Loaded, Use basemodel")
     else:
         url = f'https://huggingface.co/{lora_add}/tree/main'
-        lora_name = scrape_lora_link(url)    
-        pipe.load_lora_weights(lora_add, weight_name=lora_name)
-        pipe.fuse_lora(lora_scale=lora_scale)
-        gr.Info(f"{lora_add} Loaded")
+        lora_name = scrape_lora_link(url)
+        if lora_name:
+            pipe.load_lora_weights(lora_add, weight_name=lora_name)
+            pipe.fuse_lora(lora_scale=lora_scale)
+            gr.Info(f"{lora_add} Loaded")
+        else:
+            gr.Warning(f"{lora_add} Load fail, check again.")
 
 @spaces.GPU()
 def generate_image(
@@ -118,7 +121,7 @@ examples = [
 
 with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
     gr.HTML("<h1><center>Flux Labs</center></h1>")
-    gr.HTML("<p><center>Choose the LoRA model on the right menu</center></p>")
+    gr.HTML("<p><center>Load the LoRA model on the menu</center></p>")
     with gr.Row():
         with gr.Column(scale=4):
             img = gr.Gallery(label='flux Generated Image', columns = 1, preview=True, height=600)