Spaces:
Runtime error
Runtime error
yiyixuxu
commited on
Commit
•
18683f1
1
Parent(s):
78a010b
fixed a typo
Browse files
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 = []
|