wicked-wizard commited on
Commit
b5b41ec
·
verified ·
1 Parent(s): 2f1fdb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -145,7 +145,7 @@ class AnimateController:
145
  pose_tensor = torch.stack(pose_tensor_list, dim=0) # (f, c, h, w)
146
  pose_tensor = pose_tensor.transpose(0, 1)
147
  pose_tensor = pose_tensor.unsqueeze(0)
148
- video = torch.cat([ref_image_tensor, pose_tensor, video], dim=0)
149
 
150
  save_dir = f"./output/gradio"
151
  if not os.path.exists(save_dir):
@@ -156,7 +156,7 @@ class AnimateController:
156
  save_videos_grid(
157
  video,
158
  out_path,
159
- n_rows=3,
160
  fps=src_fps,
161
  )
162
 
 
145
  pose_tensor = torch.stack(pose_tensor_list, dim=0) # (f, c, h, w)
146
  pose_tensor = pose_tensor.transpose(0, 1)
147
  pose_tensor = pose_tensor.unsqueeze(0)
148
+ # video = torch.cat([ref_image_tensor, pose_tensor, video], dim=0)
149
 
150
  save_dir = f"./output/gradio"
151
  if not os.path.exists(save_dir):
 
156
  save_videos_grid(
157
  video,
158
  out_path,
159
+ n_rows=1,
160
  fps=src_fps,
161
  )
162