Update app.py

app.py

@@ -24,21 +24,25 @@ def report_feedback():
         return "Feedback recorded. Thank you!"
     return "No prediction available to report."
 
-# Define the Gradio interface for prediction
-with gr.Blocks() as demo:
-    gr.Markdown("### Vision Transformer Model")
-    gr.Markdown("Upload an image to classify it using the Vision Transformer model.")
-    
-    image_input = gr.Image(type="pil", label="Upload Image")
-    prediction_output = gr.Textbox(label="Prediction")
-    confidence_output = gr.Textbox(label="Confidence")
-    
-    submit_btn = gr.Button("Submit")
-    feedback_btn = gr.Button("The model was wrong")
-    
-    submit_btn.click(predict, inputs=image_input, outputs=[prediction_output, confidence_output])
-    feedback_btn.click(report_feedback)
+# Define the Gradio interface for prediction and feedback
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("### Vision Transformer Model")
+        gr.Markdown("Upload an image to classify it using the Vision Transformer model.")
+        
+        image_input = gr.Image(type="pil", label="Upload Image")
+        prediction_output = gr.Textbox(label="Prediction", interactive=False)
+        confidence_output = gr.Textbox(label="Confidence", interactive=False)
+        feedback_output = gr.Textbox(label="Feedback Status", interactive=False)
+        
+        submit_btn = gr.Button("Submit")
+        feedback_btn = gr.Button("The model was wrong")
+        
+        submit_btn.click(predict, inputs=image_input, outputs=[prediction_output, confidence_output])
+        feedback_btn.click(report_feedback, outputs=feedback_output)
+
+    # Launch the Gradio interface
+    demo.launch(share=True)
 
-# Launch the Gradio interface
 if __name__ == "__main__":
-    demo.launch(share=True)
+    main()