Update app.py
Browse filesUpdated 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}
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-
|
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():
|