improve model for init stage

app.py

@@ -1,4 +1,11 @@
 import gradio as gr
+import pandas as pd
+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())
 
 
 def predict(*args):
@@ -31,14 +38,16 @@ with gr.Blocks() as demo:
 
             home_team_id = gr.Dropdown(
                 label="Home Team",
-                choices=["zeel", "Ravi", "Sudip", "sai"],
+                choices=home_team_id,
+                value=lambda: random.choice(home_team_id),
             )
 
         with gr.Column(): 
 
             Away_team_id = gr.Dropdown(
                 label="Away Team",
-                choices=["zeel", "Ravi", "Sudip", "sai"],
+                choices=away_team_id,
+                value=lambda: random.choice(away_team_id),
             )
 
     with gr.Row():

merged_all_table.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f92279220485363b8c4159144a63a2580bbe523256cce4f240064f294809cbb
+size 11098355