Update functions.py

functions.py

@@ -44,8 +44,8 @@ def prediction(model, X):
   predictions = np.argmax(y_pred, axis=1)
   for index, val in enumerate(predictions):
     if val == 0:
-      print(f"Text {index} indicates the person is feeling FEAR")
+      return f"Text {index} indicates the person is feeling FEAR"
     elif val == 1:
-      print(f"Text {index} indicates the person is feeling ANGER")
+      return f"Text {index} indicates the person is feeling ANGER"
     else:
-      print(f"Text {index} indicates the person is feeling JOY")
+      return f"Text {index} indicates the person is feeling JOY"