yiyixuxu commited on
Commit
18683f1
1 Parent(s): 78a010b

fixed a typo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -129,7 +129,7 @@ def run_inference(url, sampling_interval, search_query, bs=256):
129
  bs = min(n_frames,bs)
130
  print(f"extracted {n_frames} frames, now encoding images")
131
  # encoding images one batch at a time, combine all batch outputs -> image_features, size n_frames x 512
132
- image_features = torch.empty(size=(n_frames, 512).to(device)
133
  print(f"batch size :{bs} ; number of batches: {len(range(0, n_frames,bs))}")
134
  for b in range(0, n_frames,bs):
135
  images = []
 
129
  bs = min(n_frames,bs)
130
  print(f"extracted {n_frames} frames, now encoding images")
131
  # encoding images one batch at a time, combine all batch outputs -> image_features, size n_frames x 512
132
+ image_features = torch.empty(size=(n_frames, 512)).to(device)
133
  print(f"batch size :{bs} ; number of batches: {len(range(0, n_frames,bs))}")
134
  for b in range(0, n_frames,bs):
135
  images = []