F-allahmoradi commited on
Commit
e7b6607
·
verified ·
1 Parent(s): e2c5ca3

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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: 'منفی', 1: 'خنثی', 0: 'مثبت'}
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