Update vit_Training.py

vit_Training.py

@@ -69,6 +69,8 @@ class Custom_VIT_Model:
         self.df = pd.concat([self.df, new_entry], ignore_index=True)
         self.df.to_csv(self.data_file, index=False)
 
+        print(self.training_data)
+
         # Check if we have 100 images for retraining
         if len(self.df) >= 100:
             self.retrain_model()