Update functions.py

functions.py

@@ -40,12 +40,13 @@ def text_model(df):
   return df['preprocessed_text']
 
 def prediction(model, X):
-  y_pred = model.predict(X)
-  predictions = np.argmax(y_pred, axis=1)
-  for index, val in enumerate(predictions):
-    if val == 0:
-      return f"Text {index} indicates the person is feeling FEAR"
-    elif val == 1:
-      return f"Text {index} indicates the person is feeling ANGER"
-    else:
-      return f"Text {index} indicates the person is feeling JOY"
+    model = load_model('best_model.keras')
+    y_pred = model.predict(X)
+    predictions = np.argmax(y_pred, axis=1)
+    for index, val in enumerate(predictions):
+        if val == 0:
+            return f"Text {index} indicates the person is feeling FEAR"
+        elif val == 1:
+            return f"Text {index} indicates the person is feeling ANGER"
+        else:
+            return f"Text {index} indicates the person is feeling JOY"