Update app.py

app.py

@@ -47,10 +47,17 @@ with gr.Blocks() as demo:
 
     submit_btn = gr.Button("Submit")
     feedback_btn = gr.Button("The model was wrong")
+    clear_btn = gr.Button("Clear")
 
     submit_btn.click(predict, inputs=image_input, outputs=[prediction_output, confidence_output])
     feedback_btn.click(report_feedback, outputs=feedback_output)
 
+    # Clear button logic
+    def clear_all():
+        return None, "", ""
+
+    clear_btn.click(clear_all, outputs=[image_input, prediction_output, confidence_output])
+
 # Launch the Gradio interface
 if __name__ == "__main__":
     demo.launch(share=True)