Update app.py

app.py

@@ -58,18 +58,15 @@ def infer(model, replay_file,
         ot_timer = ot_time_remaining[0] - ot_time_remaining
         timer[is_ot] = -ot_timer  # Negate to indicate overtime
 
-    goal_diff = scoreboard[:, SB_BLUE_SCORE] - scoreboard[:, SB_ORANGE_SCORE]
+    goal_diff = serialized_scoreboards[:, SB_BLUE_SCORE] - serialized_scoreboards[:, SB_ORANGE_SCORE]
     goal_diff_diff = goal_diff.diff(prepend=torch.Tensor([0]))
 
-    # if nullify_goal_difference:
-    #     features.scoreboard[..., 2] = 0
-
     bs = 512
     predictions = []
     it = trange(len(serialized_states), desc="Running model")
     for i in range(0, len(serialized_states), bs):
         batch = (serialized_states[i:i + bs].clone().to(DEVICE), 
-                 scoreboard[i:i + bs].clone().to(DEVICE))
+                 serialized_scoreboards[i:i + bs].clone().to(DEVICE))
         if nullify_goal_difference:
             batch[:, SB_BLUE_SCORE] = 0
             batch[:, SB_ORANGE_SCORE] = 0