shaheerawan3 commited on
Commit
65778eb
·
verified ·
1 Parent(s): e89405a

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -91,7 +91,7 @@ def train_model():
91
  save_strategy="epoch", # Save model after each epoch
92
  load_best_model_at_end=True, # Load best model at end of training
93
  push_to_hub=True, # Push model to Hugging Face Hub
94
- hub_model_id="your-username/sentiment-analysis-model" # Replace with your username
95
  )
96
 
97
  # Step 7: Initialize the trainer
 
91
  save_strategy="epoch", # Save model after each epoch
92
  load_best_model_at_end=True, # Load best model at end of training
93
  push_to_hub=True, # Push model to Hugging Face Hub
94
+ hub_model_id="shaheerawan3/Vibescribe" # Replace with your username
95
  )
96
 
97
  # Step 7: Initialize the trainer