Update main.py

main.py

@@ -177,8 +177,9 @@ def modelsummary(data):
 def modelpredict(data):
     data = txt_preprocess(data)
     pred = model.predict(pd.Series([data]))
-    print(pred)
-    return pred
+    labelpred = np.argmax(pred,axis=0)
+    dict_labels={"label":original_values(labelpred)}
+    return dict_labels
     # return {"anger": .90, "happy": .02, "emotionlabel": "anger"}
     
 #OTP code