Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ def infer(model, replay_file,
|
|
64 |
goal_diff = serialized_scoreboards[:, SB_BLUE_SCORE] - serialized_scoreboards[:, SB_ORANGE_SCORE]
|
65 |
goal_diff_diff = goal_diff.diff(prepend=torch.Tensor([0]))
|
66 |
|
67 |
-
bs =
|
68 |
predictions = []
|
69 |
it = trange(len(serialized_states), desc="Running model")
|
70 |
for i in range(0, len(serialized_states), bs):
|
@@ -317,5 +317,5 @@ with TemporaryDirectory() as temp_dir:
|
|
317 |
show_progress="full",
|
318 |
)
|
319 |
|
320 |
-
demo.queue(default_concurrency_limit=None)
|
321 |
-
demo.launch()
|
|
|
64 |
goal_diff = serialized_scoreboards[:, SB_BLUE_SCORE] - serialized_scoreboards[:, SB_ORANGE_SCORE]
|
65 |
goal_diff_diff = goal_diff.diff(prepend=torch.Tensor([0]))
|
66 |
|
67 |
+
bs = 512
|
68 |
predictions = []
|
69 |
it = trange(len(serialized_states), desc="Running model")
|
70 |
for i in range(0, len(serialized_states), bs):
|
|
|
317 |
show_progress="full",
|
318 |
)
|
319 |
|
320 |
+
demo.queue(default_concurrency_limit=None)
|
321 |
+
demo.launch()
|