Update app.py

app.py

@@ -6,6 +6,8 @@ from vit_Training import Custom_VIT_Model
 # Initialize the model
 model = CustomModel()
 
+model_training = Custom_VIT_Model()
+
 # Variable to store the last prediction result
 last_prediction = None
 
@@ -23,7 +25,7 @@ def report_feedback():
         correct_label = 1 if predicted_label == 0 else 0  # Invert the label
         print(f"Reporting feedback: predicted_label={predicted_label}, correct_label={correct_label}")  # Debugging line
         try:
-            model.add_data(image, correct_label)  # Pass the incorrect prediction to the model
+            model_training.add_data(image, correct_label)  # Pass the incorrect prediction to the model
             print("Feedback recorded successfully.")  # Debugging line
             return "Feedback recorded. Thank you!"
         except Exception as e: