Update app.py

app.py

@@ -8,18 +8,15 @@ home_team_id = sorted(data["home_team_long_name"].unique())
 away_team_id = sorted(data["away_team_long_name"].unique())
 
 
-def predict(*args):
-    # TODO: need to complete the function
-    # if str(home_team_id.value) == None:
-    #     print(str(home_team_id.value))
-    #     raise gr.Error("Home Team is required")
+def predict(Home_team, Away_team):
+
+    if Home_team == "":
+        raise gr.Error("Home Team is required")
     
-    # if str(Away_team_id.value)  == None:
-    #     raise gr.Error("Away Team is required")
+    if Away_team  == "":
+        raise gr.Error("Away Team is required")
 
-    # print(str(home_team_id.value))
-    # return str(home_team_id.value)
-    return "None"
+    return "Model is in under construction 🛠️🚜"
 
 # markup table for markdown
 # # Members:
@@ -46,18 +43,20 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
 
-            home_team_id = gr.Dropdown(
+            dd_home_team = gr.Dropdown(
                 label="Home Team",
                 choices=home_team_id,
-                max_choices=1,
+                info="Select Your Home Team:",
+                multiselect=False,
             )
 
         with gr.Column(): 
-
-            Away_team_id = gr.Dropdown(
+        
+            dd_away_team = gr.Dropdown(
                 label="Away Team",
                 choices=away_team_id,
-                max_choices=1,
+                info="Select Your Away Team:",
+                multiselect=False,
             )
 
     with gr.Row():
@@ -69,8 +68,8 @@ with gr.Blocks() as demo:
     predict_btn.click(
         predict,
         inputs=[
-            home_team_id,
-            Away_team_id,
+            dd_home_team,
+            dd_away_team,
         ],
         outputs=[Answer],
     )