Spaces:
Running
Running
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -258,7 +258,7 @@ class Predictor:
|
|
258 |
frame_count = len(video_frames)
|
259 |
transformed_video_frames = []
|
260 |
|
261 |
-
batch_shape = (batch_size
|
262 |
frames = np.zeros(batch_shape, dtype=np.uint8)
|
263 |
frame_idx = 0
|
264 |
|
|
|
258 |
frame_count = len(video_frames)
|
259 |
transformed_video_frames = []
|
260 |
|
261 |
+
batch_shape = (batch_size) + video_frames[0].shape
|
262 |
frames = np.zeros(batch_shape, dtype=np.uint8)
|
263 |
frame_idx = 0
|
264 |
|