Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -365,7 +365,7 @@ def generate(prompt,
|
|
365 |
# We expect B=1 from typical generation
|
366 |
video_single_batch = result_frames_tensor[0] # Shape: (C, F, H, W)
|
367 |
video_single_batch = (video_single_batch / 2 + 0.5).clamp(0, 1) # Normalize to [0,1]
|
368 |
-
video_single_batch = video_single_batch.permute(1, 2, 3, 0).cpu().numpy() # F, H, W, C
|
369 |
|
370 |
for frame_idx in range(video_single_batch.shape[0]):
|
371 |
frame_np = (video_single_batch[frame_idx] * 255).astype(np.uint8)
|
|
|
365 |
# We expect B=1 from typical generation
|
366 |
video_single_batch = result_frames_tensor[0] # Shape: (C, F, H, W)
|
367 |
video_single_batch = (video_single_batch / 2 + 0.5).clamp(0, 1) # Normalize to [0,1]
|
368 |
+
video_single_batch = video_single_batch.permute(1, 2, 3, 0).cpu().float().numpy() # F, H, W, C
|
369 |
|
370 |
for frame_idx in range(video_single_batch.shape[0]):
|
371 |
frame_np = (video_single_batch[frame_idx] * 255).astype(np.uint8)
|