Spaces:
Sleeping
Sleeping
Update vit_Training.py
Browse files- vit_Training.py +2 -2
vit_Training.py
CHANGED
@@ -77,7 +77,7 @@ class Custom_VIT_Model:
|
|
77 |
# Print the current state of the training data for debugging
|
78 |
print("Current training data:")
|
79 |
print(self.df)
|
80 |
-
|
81 |
|
82 |
|
83 |
|
@@ -85,7 +85,7 @@ class Custom_VIT_Model:
|
|
85 |
if len(self.df) >= 100:
|
86 |
print("Retraining the model as we have enough data.")
|
87 |
self.retrain_model()
|
88 |
-
|
89 |
|
90 |
|
91 |
|
|
|
77 |
# Print the current state of the training data for debugging
|
78 |
print("Current training data:")
|
79 |
print(self.df)
|
80 |
+
|
81 |
|
82 |
|
83 |
|
|
|
85 |
if len(self.df) >= 100:
|
86 |
print("Retraining the model as we have enough data.")
|
87 |
self.retrain_model()
|
88 |
+
|
89 |
|
90 |
|
91 |
|