Update app.py

app.py

@@ -4,8 +4,8 @@ import random
 
 data = pd.read_pickle("merged_all_table.pkl", compression='bz2')
 
-home_team_id = list(data["home_team_api_id"].unique())
-away_team_id = list(data["away_team_api_id"].unique())
+home_team_id = list(data["team_long_name"].unique())
+away_team_id = list(data["team_long_name"].unique())
 
 
 def predict(*args):
@@ -39,6 +39,7 @@ with gr.Blocks() as demo:
             home_team_id = gr.Dropdown(
                 label="Home Team",
                 choices=home_team_id,
+                max_choices=1,
                 # value=lambda: random.choice(home_team_id),
             )
 
@@ -47,6 +48,7 @@ with gr.Blocks() as demo:
             Away_team_id = gr.Dropdown(
                 label="Away Team",
                 choices=away_team_id,
+                max_choices=1,
                 # value=lambda: random.choice(away_team_id),
             )