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 |
|