Spaces:
Paused
Paused
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)
|