Spaces:
Running
on
Zero
Running
on
Zero
Fix frame sampling scheme.
Browse files
app.py
CHANGED
@@ -163,7 +163,7 @@ def generate(image, video, state, state_, textbox_in, temperature, top_p, max_ou
|
|
163 |
tensor.append(process_image(image, processor).to(handler.model.device, dtype=dtype))
|
164 |
modals.append('IMAGE')
|
165 |
if not os.path.exists(image) and os.path.exists(video):
|
166 |
-
tensor.append(process_video(video, processor, num_frames=num_frames, sample_scheme='
|
167 |
modals.append('VIDEO')
|
168 |
if os.path.exists(image) and os.path.exists(video):
|
169 |
raise NotImplementedError("Not support image and video at the same time")
|
|
|
163 |
tensor.append(process_image(image, processor).to(handler.model.device, dtype=dtype))
|
164 |
modals.append('IMAGE')
|
165 |
if not os.path.exists(image) and os.path.exists(video):
|
166 |
+
tensor.append(process_video(video, processor, num_frames=num_frames, sample_scheme='uniform').to(handler.model.device, dtype=dtype))
|
167 |
modals.append('VIDEO')
|
168 |
if os.path.exists(image) and os.path.exists(video):
|
169 |
raise NotImplementedError("Not support image and video at the same time")
|