Update app.py

app.py

@@ -108,6 +108,7 @@ def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler=
     except Exception as e:
         print(f"Error when trying to open the image: {e}")
         return None
+        
 def clear():
   return None
     
@@ -176,12 +177,17 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                             
             with gr.Row():
                 text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
-            with gr.Row():                
-                clr_button =gr.Button("Clear",variant="primary", elem_id="clear_button")
-                clr_button.click(lambda: gr.Textbox(value=""), None, text_prompt) 
-                
             with gr.Row():
                 image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
+                
+            with gr.Row(): 
+                clear_btn = gr.Button(value="Clear",variant="primary", elem_id="clear_button")
+                clear_btn.click(clear, inputs=[], outputs=[image_output])
+                #clr_button =gr.Button("Clear",variant="primary", elem_id="clear_button")
+                #clr_button.click(lambda: gr.Textbox(value=""), None, text_prompt) 
+                
+            #with gr.Row():
+                #image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
 
             gr.Examples(
                 examples = examples,