Update prediction.py

prediction.py

@@ -155,9 +155,9 @@ def run():
       y_pred_df = pd.DataFrame(y_pred_conv, columns=['0', '1', '2'])
       y_pred_final=y_pred_df.idxmax(1).astype(int)
 
-      if y_pred_final == 2:
+      if y_pred_final.any() == 2:
             st.write('## Dude, your guest gave Positive feedback')
-      if y_pred_final == 1:
+      if y_pred_final.any() == 1:
             st.write('## Dude, your guest gave Neutral feedback')
       else:
             st.write('## Attention, your guest gave Negative feedback')