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
|