Spaces:
Runtime error
Runtime error
Update train.py
Browse files
train.py
CHANGED
@@ -82,4 +82,8 @@ print("✅ Training complete.")
|
|
82 |
# STEP 9: Save model AND tokenizer
|
83 |
trainer.save_model("./results")
|
84 |
tokenizer.save_pretrained("./results")
|
85 |
-
|
|
|
|
|
|
|
|
|
|
82 |
# STEP 9: Save model AND tokenizer
|
83 |
trainer.save_model("./results")
|
84 |
tokenizer.save_pretrained("./results")
|
85 |
+
|
86 |
+
# Push to Hugging Face Hub
|
87 |
+
model.push_to_hub("eemm-deberta-v3-small")
|
88 |
+
tokenizer.push_to_hub("eemm-deberta-v3-small")
|
89 |
+
print("✅ Model and tokenizer pushed to Hugging Face Hub.")
|