Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
|
@@ -49,7 +49,7 @@ def predict_sentiment(batch):
|
|
| 49 |
return {'sentiment': predictions.cpu()}
|
| 50 |
|
| 51 |
# Mapping sentiment labels
|
| 52 |
-
sentiment_labels_en = { 2: 'منفی',
|
| 53 |
|
| 54 |
|
| 55 |
# Adding sentiment prediction to tokenized dataset
|
|
|
|
| 49 |
return {'sentiment': predictions.cpu()}
|
| 50 |
|
| 51 |
# Mapping sentiment labels
|
| 52 |
+
sentiment_labels_en = { 2: 'منفی', 0: 'خنثی', 1: 'مثبت'}
|
| 53 |
|
| 54 |
|
| 55 |
# Adding sentiment prediction to tokenized dataset
|