Spaces:
Configuration error
Configuration error
Update app.py
Browse files
app.py
CHANGED
@@ -24,6 +24,7 @@ def compute_batch_hashes(vid_path):
|
|
24 |
h_prev = None
|
25 |
batch = []
|
26 |
for i in range(0, len(vr), DOWNSAMPLE * BATCH_SIZE):
|
|
|
27 |
ids = [id for id in range(i, min(i + DOWNSAMPLE * BATCH_SIZE, len(vr)), DOWNSAMPLE)]
|
28 |
vr.seek(0)
|
29 |
batch = jnp.array(vr.get_batch(ids).asnumpy())
|
|
|
24 |
h_prev = None
|
25 |
batch = []
|
26 |
for i in range(0, len(vr), DOWNSAMPLE * BATCH_SIZE):
|
27 |
+
print(f"batch_{i}")
|
28 |
ids = [id for id in range(i, min(i + DOWNSAMPLE * BATCH_SIZE, len(vr)), DOWNSAMPLE)]
|
29 |
vr.seek(0)
|
30 |
batch = jnp.array(vr.get_batch(ids).asnumpy())
|