Ozziejoe commited on
Commit
ac45da5
·
verified ·
1 Parent(s): 63fe74a

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +5 -1
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
- print("✅ Model and tokenizer saved to ./results")
 
 
 
 
 
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.")