Rolv-Arild commited on
Commit
eb781a5
·
verified ·
1 Parent(s): 39fa5ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = 900
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()