Update app.py

app.py

@@ -8,6 +8,24 @@ model = CustomModel()
 
 model_training = Custom_VIT_Model()
 
+
+custom_css = """
+<style>
+#submit_btn {
+    background-color: #4CAF50; /* Green */
+    color: white;
+}
+#feedback_btn {
+    background-color: #f44336; /* Red */
+    color: white;
+}
+#clear_btn {
+    background-color: #2196F3; /* Blue */
+    color: white;
+}
+</style>
+"""
+
 # Variable to store the last prediction result
 last_prediction = None
 
@@ -60,22 +78,7 @@ 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])
 
-custom_css = """
-<style>
-#submit_btn {
-    background-color: #4CAF50; /* Green */
-    color: white;
-}
-#feedback_btn {
-    background-color: #f44336; /* Red */
-    color: white;
-}
-#clear_btn {
-    background-color: #2196F3; /* Blue */
-    color: white;
-}
-</style>
-"""
+
     
 
 # Launch the Gradio interface