Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
|
@@ -53,11 +53,11 @@ def predict_sentiment(batch):
|
|
| 53 |
|
| 54 |
# Mapping sentiment labels
|
| 55 |
sentiment_labels = {
|
| 56 |
-
0: '
|
| 57 |
-
1: '
|
| 58 |
-
2: '
|
| 59 |
-
3: '
|
| 60 |
-
4: '
|
| 61 |
}
|
| 62 |
|
| 63 |
# Adding sentiment prediction to tokenized dataset
|
|
|
|
| 53 |
|
| 54 |
# Mapping sentiment labels
|
| 55 |
sentiment_labels = {
|
| 56 |
+
0: 'بسیار عصیانی',
|
| 57 |
+
1: 'عصبانی',
|
| 58 |
+
2: ' خنثی',
|
| 59 |
+
3: 'مثبت',
|
| 60 |
+
4: ' بسیار مثبت'
|
| 61 |
}
|
| 62 |
|
| 63 |
# Adding sentiment prediction to tokenized dataset
|