Update app.py

app.py

@@ -249,21 +249,12 @@ DESCRIPTION = """
 # Next Goal Predictor
 Upload a replay file to get a plot of the next goal prediction.
 
-The model is trained on about 50,000 hours of SSL and RLCS replays in 1s, 2s, and 3s.
-
-It predicts the probability of each team scoring the next goal, in addition to ties.<br>
-It also predicts the probability that each team will score in 1s, 2s, etc. up to 60+ seconds.<br>
-The plot only shows the team predictions, but you can download the full predictions if you want. 
+The model is trained on about 14 000 hours of SSL and RLCS replays in 1v1, 2v2, and 3v3 using [this dataset](https://www.kaggle.com/datasets/rolvarild/high-level-rocket-league-replay-dataset).<br>
+It predicts the probability that each team will score at 1 second intervals up to 60+ seconds.
+It also predicts ties (ball hitting the ground at 0s)<br>
+The plot only shows the totals for each team, but you can download the full predictions if you want. 
 """.strip()
 
-# RADIO_OPTIONS = [
-#     ("**Default**<br>Uses the model as it is trained, with no modifications.", "default"),
-#     (
-#     "**Nullify goal difference**<br>Makes the model think the goal difference is always 0, so it doesn't have a bias towards one team.",
-#     "nullify_goal_difference"),
-#     ("**Ignore ties**<br>Makes the model pretend every situation is an overtime (e.g. ties are impossible).",
-#      "ignore_ties"),
-# ]
 RADIO_OPTIONS = ["Default", "Nullify goal difference", "Ignore ties"]
 RADIO_INFO = """
 - **Default**: Uses the model as it is trained, with no modifications.
@@ -282,7 +273,7 @@ def gradio_app():
             # Use gr.Column to stack components vertically
             with gr.Column():
                 file_input = gr.File(label="Upload Replay File", type="filepath", file_types=[".replay"])
-                checkboxes = gr.Radio(label="Options", choices=RADIO_OPTIONS, type="index", value=0,
+                checkboxes = gr.Radio(label="Options", choices=RADIO_OPTIONS, type="index", value=RADIO_OPTIONS[0],
                                       info=RADIO_INFO)
                 submit_button = gr.Button("Generate Predictions")
                 plot_output = gr.Plot(label="Predictions")