Update vit_Training.py

vit_Training.py

@@ -78,6 +78,8 @@ 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.")