Ani-404 commited on
Commit
0194fbd
·
1 Parent(s): 7cae94c

uppdated training script

Browse files
Files changed (1) hide show
  1. finance/train_finbert.py +1 -4
finance/train_finbert.py CHANGED
@@ -105,10 +105,7 @@ def main():
105
  print(f"Training complete. Saving final model to '{output_dir}'...")
106
  trainer.save_model(output_dir)
107
  tokenizer.save_pretrained(output_dir)
108
- print("-----" * 10)
109
- print("SUCCESS: Model saved to your Google Drive in the 'Colab_Data' folder.")
110
- print("You can now download the 'finbert_emotion_model' folder and place it in your project's 'finance/' directory.")
111
- print("-----" * 10)
112
 
113
  if __name__ == "__main__":
114
  main()
 
105
  print(f"Training complete. Saving final model to '{output_dir}'...")
106
  trainer.save_model(output_dir)
107
  tokenizer.save_pretrained(output_dir)
108
+ print("Model and tokenizer saved successfully.")
 
 
 
109
 
110
  if __name__ == "__main__":
111
  main()