Update pages/2_play_game.py

pages/2_play_game.py

@@ -5,6 +5,13 @@ import numpy as np
 import tensorflow as tf
 import random
 from tensorflow.keras.models import load_model
+import os
+
+USER = os.environ['USER']
+PASSWORD = os.environ['PASSWORD']
+ACCOUNT = os.environ['ACCOUNT']
+WAREHOUSE = os.environ['WAREHOUSE']
+DATABASE = os.environ['DATABASE']
 
 
 def on_page_load():
@@ -32,7 +39,7 @@ def get_away_team(cnx, query_string):
     return players    
               
 def find_away_team():
-    cnx = snowflake.connector.connect(**st.secrets["snowflake"])
+    cnx = snowflake.connector.connect(user=USER, password=PASSWORD, account=ACCOUNT, warehouse=WAREHOUSE, database=DATABASE)
 
     data = get_away_team(cnx, query_string)
     cnx.close()