Rolv-Arild commited on
Commit
59ac7e2
·
verified ·
1 Parent(s): eead613

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,8 +70,8 @@ def infer(model, replay_file,
70
  if nullify_goal_difference:
71
  batch[:, SB_BLUE_SCORE] = 0
72
  batch[:, SB_ORANGE_SCORE] = 0
73
- out = model(batch)
74
- it.update(len(batch))
75
 
76
  predictions.append(out)
77
 
 
70
  if nullify_goal_difference:
71
  batch[:, SB_BLUE_SCORE] = 0
72
  batch[:, SB_ORANGE_SCORE] = 0
73
+ out = model(*batch)
74
+ it.update(len(batch[0]))
75
 
76
  predictions.append(out)
77