Update app.py

app.py

@@ -56,11 +56,12 @@ def scrape_lora_link(url):
     except requests.RequestException as e:
         raise gr.Error(f"An error occurred while fetching the URL: {e}")
 
-def enable_lora(lora_scale,lora_add):
+def enable_lora(lora_scale,lora_add,progress=gr.Progress(track_tqdm=True)):
+    pipe.unload_lora_weights()
     if not lora_add:
         gr.Info("No Lora Loaded, Use basemodel")
+        return gr.update(value="")
     else:
-        pipe.unload_lora_weights()
         url = f'https://huggingface.co/{lora_add}/tree/main'
         lora_name = scrape_lora_link(url)
         if lora_name:
@@ -68,11 +69,13 @@ def enable_lora(lora_scale,lora_add):
             pipe.load_lora_weights(lora_add, weight_name=lora_name)
             pipe.fuse_lora(lora_scale=lora_scale)
             gr.Info(f"{lora_add} Loaded")
+            return gr.update(label="LoRA Loaded Now")
         else:
             try:
                 pipe.load_lora_weights(lora_add)
                 pipe.fuse_lora(lora_scale=lora_scale)
                 gr.Info(f"{lora_add} Loaded")
+                return gr.update(label="LoRA Loaded Now")
             except:
                 raise gr.Error(f"{lora_add} Load fail, check again.")
 
@@ -186,10 +189,9 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
                     value=1.0,
                 )
                 lora_add = gr.Textbox(
-                    label="Add Flux LoRA",
+                    label="Flux LoRA",
                     info="Copy the HF LoRA model name here",
                     lines=1,
-                    value="Shakker-Labs/AWPortrait-FL",
                 )
                 lora_word = gr.Textbox(
                     label="Add Flux LoRA Trigger Word",
@@ -206,7 +208,7 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
         examples_per_page=4,
     )
 
-    load_lora.click(fn=enable_lora, inputs=[lora_scale,lora_add])
+    load_lora.click(fn=enable_lora, inputs=[lora_scale,lora_add], outputs=lora_add)
     
     gr.on(
         triggers=[