Update app.py

app.py

@@ -154,17 +154,27 @@ with gr.Blocks() as demo:
         with gr.Column():
             output = gr.Markdown()
 
-    # Update the example image when an example is selected
+    
     example_selection.change(
         lambda selected_example: load_example(selected_example),
         inputs=[example_selection],
         outputs=[example_display]
     )
 
-    # Submit button to trigger inference
+    
+    clear_button = gr.Button("Clear Example")
+
+    
+    clear_button.click(
+        lambda: None,  
+        inputs=[],
+        outputs=[example_display]
+    )
+
+    
     submit_button = gr.Button("Submit")
 
-    # Handle submit button click
+   
     submit_button.click(
         lambda user_image, selected_example: inference(user_image=user_image, selected_example=selected_example),
         inputs=[inp_image, example_selection],