Update app.py

app.py

@@ -29,6 +29,7 @@ def run_lora(prompt, weight):
     print("Query function executed successfully.")
     return Image.open(image_bytes)
 
+# Gradio UI
 print("Before Gradio Interface")
 
 title = gr.HTML("<h1>LoRA the Explorer</h1>")
@@ -37,16 +38,17 @@ gallery = gr.Gallery(
     [(item["image"], item["title"]) for item in loras],
     label="LoRA Gallery",
     allow_preview=False,
-    columns=3
+    columns=3,
+    name="gallery"  # Add a name for the gallery component
 )
 
-prompt = gr.Textbox(label="Prompt", lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
+prompt = gr.Textbox(label="Prompt", lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA", name="prompt")
 
-advanced_options = gr.Accordion("Advanced options", open=False)
+advanced_options = gr.Accordion("Advanced options", open=False, name="advanced_options")
 
-weight = gr.Slider(0, 10, value=1, step=0.1, label="LoRA weight")
+weight = gr.Slider(0, 10, value=1, step=0.1, label="LoRA weight", name="weight")
 
-result = gr.Image(interactive=False, label="Generated Image")
+result = gr.Image(interactive=False, label="Generated Image", name="result_image")
 
 gr.Interface(
     fn=run_lora,
@@ -55,4 +57,4 @@ gr.Interface(
     css="custom.css"  # Make sure your custom CSS file is in the same directory
 ).launch()
 
-print("After Gradio Interface")
+print("After Gradio Interface")