Update app.py

app.py

@@ -56,21 +56,27 @@ def report_feedback():
         return "No prediction available to report."
 
 # Define the Gradio interface
-with gr.Blocks(css=custom_css,theme=gr.themes.Soft()) as demo:
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) 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)
 
+    # Create a row for prediction and confidence outputs
+    with gr.Row():
+        prediction_output = gr.Textbox(label="Prediction", interactive=False)
+        confidence_output = gr.Textbox(label="Confidence", interactive=False)
+
+    # Create a row for feedback output
+    feedback_output = gr.Textbox(label="Feedback Status", interactive=False)
+    
+    # Buttons
     with gr.Row():
         submit_btn = gr.Button("Submit", variant="primary", elem_id="submit_btn")
         feedback_btn = gr.Button("The model was wrong", variant="secondary", elem_id="feedback_btn")
         
-    gr.Markdown("<br>")
-    clear_btn = gr.Button("Clear", elem_id="clear_btn")
+    gr.Markdown("<br>")  # Adding space below buttons
+    clear_btn = gr.Button("Clear", elem_id="clear_btn", elem_classes="clear-button")  # Add class to clear button
 
     submit_btn.click(predict, inputs=image_input, outputs=[prediction_output, confidence_output])
     feedback_btn.click(report_feedback, outputs=feedback_output)
@@ -81,9 +87,5 @@ with gr.Blocks(css=custom_css,theme=gr.themes.Soft()) as demo:
 
     clear_btn.click(clear_all, outputs=[image_input, prediction_output, confidence_output, feedback_output])
 
-
-    
-
-# Launch the Gradio interface
 if __name__ == "__main__":
     demo.launch(share=True)