Update models.py

models.py

@@ -34,10 +34,17 @@ def modelsummary(data):
     return { 'summary':result}
 
 
+  #   def modelpredict(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 {"emotion":map[np.argmax(pred,axis=-1)[0]]}
 #Sentiment model
 def modelpredict(data):
-    data = txt_preprocess(data)
+    map = {0:'anger', 1:'sadness', 2:'joy', 3:'surprise', 4:'love', 5:'sympathy', 6:'fear'}
+    data = Cleaning.txt_preprocess(data)
     pred = model_sentiment.predict(pd.Series([data]))
-    return {'anger': float(pred[0][0]), 'sadness': float(pred[0][1]), 'joy': float(pred[0][2]), 'surprise': float(pred[0][3]),
-            'love': float(pred[0][4]), 'sympathy': float(pred[0][5]), 'fear': float(pred[0][6])}
+    return {"label":map[np.argmax(pred,axis=-1)[0]]}
+    # return {'anger': float(pred[0][0]), 'sadness': float(pred[0][1]), 'joy': float(pred[0][2]), 'surprise': float(pred[0][3]),
+    #         'love': float(pred[0][4]), 'sympathy': float(pred[0][5]), 'fear': float(pred[0][6])}