1inkusFace commited on
Commit
c8ff766
·
verified ·
1 Parent(s): b113647

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)