Update app.py

app.py

@@ -4,8 +4,9 @@ from tempfile import TemporaryDirectory
 import numpy as np
 import pandas as pd
 import torch
+from huggingface_hub import Repository
 from rlgym_tools.rocket_league.misc.serialize import serialize_game_state, serialize_scoreboard, \
-    RF_SCOREBOARD, RF_EPISODE_SECONDS_REMAINING, SB_GAME_TIMER_SECONDS, SB_BLUE_SCORE, SB_ORANGE_SCORE
+    SB_GAME_TIMER_SECONDS, SB_BLUE_SCORE, SB_ORANGE_SCORE
 from rlgym_tools.rocket_league.replays.convert import replay_to_rlgym
 from rlgym_tools.rocket_league.replays.parsed_replay import ParsedReplay
 from tqdm import trange, tqdm
@@ -14,8 +15,6 @@ os.chmod("/usr/local/lib/python3.10/site-packages/rlgym_tools/rocket_league/repl
 
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-from huggingface_hub import Repository
-
 repo = Repository(local_dir="vortex-ngp", clone_from="Rolv-Arild/vortex-ngp", token=os.getenv("HF_TOKEN"))
 repo.git_pull()
 
@@ -252,15 +251,18 @@ 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.
-It also predicts the probability that each team will score in 1s, 2s, etc. up to 60+ seconds.
-
-The plot only shows the team predictions, but you can download the full predictions. 
+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. 
 """.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.",
-    "**Ignore ties**<br>Makes the model pretend every situation is an overtime (e.g. ties are impossible)."
+    "nullify_goal_difference"),
+    ("**Ignore ties**<br>Makes the model pretend every situation is an overtime (e.g. ties are impossible).",
+     "ignore_ties"),
 ]
 
 
@@ -274,15 +276,15 @@ 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")
+                checkboxes = gr.Radio(label="Options", choices=RADIO_OPTIONS, type="index", value=0)
                 submit_button = gr.Button("Generate Predictions")
                 plot_output = gr.Plot(label="Predictions")
                 download_button = gr.DownloadButton("Download Predictions", visible=False)
 
             # Make plot on button click
             def make_plot(replay_file, radio_option, progress=gr.Progress(track_tqdm=True)):
-                nullify_goal_difference = radio_option == 0
-                ignore_ties = radio_option == 1
+                nullify_goal_difference = radio_option == 1
+                ignore_ties = radio_option == 2
                 print(f"Processing file: {replay_file}")
 
                 replay_stem = os.path.splitext(os.path.basename(replay_file))[0]