Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|