Update app.py

app.py

@@ -26,10 +26,11 @@ def infer(model, replay_file,
           nullify_goal_difference=False,
           predict_with_all_quadrants=True,
           ignore_ties=False):
-    swap_team_idx = torch.arange(model.num_outputs)
-    mid = model.num_outputs // 2
+    num_outputs = 123
+    swap_team_idx = torch.arange(num_outputs)
+    mid = num_outputs // 2
     swap_team_idx[mid:-1] = swap_team_idx[:mid]
-    swap_team_idx[:mid] += model.num_outputs // 2
+    swap_team_idx[:mid] += num_outputs // 2
 
     replay = ParsedReplay.load(replay_file)
     it = tqdm(replay_to_rlgym(replay), desc="Loading replay", total=len(replay.game_df))
@@ -73,7 +74,7 @@ def infer(model, replay_file,
     predictions = torch.cat(predictions, dim=0)
     probs = predictions.softmax(dim=-1)
 
-    bin_seconds = torch.linspace(0, 60, model.num_outputs // 2)
+    bin_seconds = torch.linspace(0, 60, num_outputs // 2)
     class_names = [
         f"{t}: {s:g}s" for t in ["Blue", "Orange"]
         for s in bin_seconds.tolist()