AfroLogicInsect commited on
Commit
e7da2c8
·
verified ·
1 Parent(s): c42c6a5

Update app.py

Browse files

Updated model to use version 2:

Final evaluation results: {'eval_loss': 0.3390733003616333, 'eval_accuracy': 0.9086666666666666, 'eval_f1': 0.9103989535644212, 'eval_precision': 0.8934531450577664, 'eval_recall': 0.928, 'eval_runtime': 9.9181, 'eval_samples_per_second': 151.239, 'eval_steps_per_second': 9.478, 'epoch': 3.0}

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  ## Load Pipeline
5
  sentiment_pipeline = pipeline("sentiment-analysis",
6
- model="AfroLogicInsect/sentiment-analysis-model")
7
 
8
  def predict_sentiment(text):
9
  if not text.strip():
 
3
 
4
  ## Load Pipeline
5
  sentiment_pipeline = pipeline("sentiment-analysis",
6
+ model="AfroLogicInsect/sentiment-analysis-model_v2")
7
 
8
  def predict_sentiment(text):
9
  if not text.strip():