Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -292,12 +292,12 @@ def generate_video(prompt: str, seed: int, image: str = None) -> tuple[str, dict
|
|
292 |
|
293 |
with torch.no_grad():
|
294 |
output = _predictor.infer(**kwargs)
|
295 |
-
|
296 |
output = (output.numpy() * 255).astype(np.uint8)
|
297 |
# Correct Transpose: (1, C, T, H, W) -> (1, T, H, W, C)
|
298 |
output = output.transpose(0, 2, 3, 4, 1)
|
299 |
output = output[0] # Remove batch dimension: (T, H, W, C)
|
300 |
-
|
301 |
|
302 |
save_dir = f"./result"
|
303 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
292 |
|
293 |
with torch.no_grad():
|
294 |
output = _predictor.infer(**kwargs)
|
295 |
+
'''
|
296 |
output = (output.numpy() * 255).astype(np.uint8)
|
297 |
# Correct Transpose: (1, C, T, H, W) -> (1, T, H, W, C)
|
298 |
output = output.transpose(0, 2, 3, 4, 1)
|
299 |
output = output[0] # Remove batch dimension: (T, H, W, C)
|
300 |
+
'''
|
301 |
|
302 |
save_dir = f"./result"
|
303 |
os.makedirs(save_dir, exist_ok=True)
|