Update vit_Training.py

vit_Training.py

@@ -78,12 +78,13 @@ class Custom_VIT_Model:
         print("Current training data:")
         print(self.df)
 
-        self.df= None
+        
 
         # Check if we have 100 images for retraining
         if len(self.df) >= 100:
             print("Retraining the model as we have enough data.")
             self.retrain_model()
+        self.df = None