F-allahmoradi commited on
Commit
802305b
·
verified ·
1 Parent(s): e768790

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +5 -5
model.py CHANGED
@@ -53,11 +53,11 @@ def predict_sentiment(batch):
53
 
54
  # Mapping sentiment labels
55
  sentiment_labels = {
56
- 0: 'Furious',
57
- 1: 'Angry',
58
- 2: 'Neutral',
59
- 3: 'Happy',
60
- 4: 'Delighted'
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