MahmoudNasser commited on
Commit
69fdfff
1 Parent(s): 268dd78

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -177,8 +177,8 @@ def modelsummary(data):
177
  def modelpredict(data):
178
  data = txt_preprocess(data)
179
  pred = model.predict(pd.Series([data]))
180
- return {'anger': float(pred[0][0]), 'sadness': float(pred[0][1]), 'joy': float(pred[0][2]), 'surprise': float(pred[0][3]),
181
- 'love': float(pred[0][4]), 'sympathy': float(pred[0][5]), 'fear': float(pred[0][6])}
182
  # return {"anger": .90, "happy": .02, "emotionlabel": "anger"}
183
 
184
  #OTP code
 
177
  def modelpredict(data):
178
  data = txt_preprocess(data)
179
  pred = model.predict(pd.Series([data]))
180
+ print(pred)
181
+ return pred
182
  # return {"anger": .90, "happy": .02, "emotionlabel": "anger"}
183
 
184
  #OTP code