Update app.py

app.py

@@ -110,7 +110,8 @@ def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler=
         return None
         
 def clear():
-  return None
+    image_output.clear()
+    return None
     
 examples = [
     "a beautiful woman with blonde hair and blue eyes",
@@ -177,15 +178,15 @@ 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():
-                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])
+                clear_btn.click(clear, inputs=[text_prompt], 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")
             #with gr.Row():
                 #image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")