Update main.py

Emotion prediction

main.py

@@ -174,13 +174,11 @@ def modelsummary(data):
     result = remove_punctuations(result)
     return { 'summary':result}
 
-def modelpredict(data):
-    data = txt_preprocess(data)
-    pred = model.predict(pd.Series([data]))
-    labelpred = np.argmax(pred,axis=0)
-    dict_labels={"label":original_values(labelpred)}
-    return dict_labels
-    # return {"anger": .90, "happy": .02, "emotionlabel": "anger"}
+def predict(data):
+  map = {0:'anger', 1:'sadness', 2:'joy', 3:'surprise', 4:'love', 5:'sympathy', 6:'fear'}
+  text = txt_preprocess(data)
+  pred=model.predict(pd.Series([data])) 
+  return map[np.argmax(pred,axis=-1)[0]]
     
 #OTP code 
 def genereteotp (email):