Hugging Face's logo

Spaces:
dylanplummer
/
Single-Rope-Contest
Running

App Files Community
dylanplummer commited on
Commit
73b1bf3
·
1 Parent(s): 0340538

fix warmup

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ else:
46
  ort_sess = ort.InferenceSession(onnx_file)
47
 
48
  print('Warmup...')
49
- dummy_input = torch.randn(1, 64, 3, 288, 288)
50
  ort_sess.run(None, {'video': dummy_input.numpy()})
51
  print('Done!')
52
 
 
46
  ort_sess = ort.InferenceSession(onnx_file)
47
 
48
  print('Warmup...')
49
+ dummy_input = torch.randn(4, 64, 3, 288, 288)
50
  ort_sess.run(None, {'video': dummy_input.numpy()})
51
  print('Done!')
52