Update app.py
Browse files
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 |
-
|
| 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=
|
| 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 |
|