Update pages/2_play_game.py

pages/2_play_game.py

@@ -83,7 +83,7 @@ if teams_good:
     winner_model = load_model('winner.keras')
 
     winner_sigmoid= winner_model.predict(winner)
-    winner_prediction = np.round(winner_sigmoid)
+    winner_prediction = np.round(winner_sigmoid[0][0])
 
     score = []
     winner_score = random.randint(90, 130)
@@ -104,7 +104,7 @@ if teams_good:
 
     box_score = pd.DataFrame(score , columns=['1', '2', '3', '4', 'Final'], index=['Home Team', 'Away Team'] )
     
-    print(f"Prediction: {winner_sigmoid}")
+    print(f"Prediction: {winner_sigmoid[0][0]}")
 
 st.markdown("<h1 style='text-align: center; color: steelblue;'>Home Team</h1>", unsafe_allow_html=True)
 st.dataframe(st.session_state.home_team_df)