Merge pull request #3 from jhj0517/feature/automatic_output
Browse files
app.py
CHANGED
|
@@ -48,16 +48,6 @@ class App:
|
|
| 48 |
|
| 49 |
with gr.Row():
|
| 50 |
btn_align_pose = gr.Button("ALIGN POSE", variant="primary")
|
| 51 |
-
with gr.Column():
|
| 52 |
-
examples = [
|
| 53 |
-
[os.path.join("assets", "videos", "dance.mp4"), os.path.join("assets", "images", "ref.png"),
|
| 54 |
-
512, 720, 0, 300]]
|
| 55 |
-
ex_step1 = gr.Examples(examples=examples,
|
| 56 |
-
inputs=[vid_dance_input, img_pose_input, nb_detect_resolution,
|
| 57 |
-
nb_image_resolution, nb_align_frame, nb_max_frame],
|
| 58 |
-
outputs=[vid_dance_output, vid_dance_output_demo],
|
| 59 |
-
fn=self.pose_alignment_infer.align_pose,
|
| 60 |
-
cache_examples="lazy")
|
| 61 |
|
| 62 |
btn_align_pose.click(fn=self.pose_alignment_infer.align_pose,
|
| 63 |
inputs=[vid_dance_input, img_pose_input, nb_detect_resolution, nb_image_resolution,
|
|
@@ -94,19 +84,6 @@ class App:
|
|
| 94 |
with gr.Row():
|
| 95 |
btn_generate = gr.Button("GENERATE", variant="primary")
|
| 96 |
|
| 97 |
-
with gr.Column():
|
| 98 |
-
examples = [
|
| 99 |
-
[os.path.join("assets", "images", "ref.png"), os.path.join("assets", "videos", "pose.mp4"),
|
| 100 |
-
"fp16", 512, 512, 300, 48, 4, 3.5, 99, 20, -1, 1]]
|
| 101 |
-
ex_step2 = gr.Examples(examples=examples,
|
| 102 |
-
inputs=[img_musepose_input, vid_pose_input, weight_dtype, nb_width, nb_height,
|
| 103 |
-
nb_video_frame_length, nb_video_slice_frame_length,
|
| 104 |
-
nb_video_slice_overlap_frame_number, nb_cfg, nb_seed, nb_steps,
|
| 105 |
-
nb_fps, nb_skip],
|
| 106 |
-
outputs=[vid_output, vid_output_demo],
|
| 107 |
-
fn=self.musepose_infer.infer_musepose,
|
| 108 |
-
cache_examples="lazy")
|
| 109 |
-
|
| 110 |
btn_generate.click(fn=self.musepose_infer.infer_musepose,
|
| 111 |
inputs=[img_musepose_input, vid_pose_input, weight_dtype, nb_width, nb_height,
|
| 112 |
nb_video_frame_length, nb_video_slice_frame_length,
|
|
|
|
| 48 |
|
| 49 |
with gr.Row():
|
| 50 |
btn_align_pose = gr.Button("ALIGN POSE", variant="primary")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
btn_align_pose.click(fn=self.pose_alignment_infer.align_pose,
|
| 53 |
inputs=[vid_dance_input, img_pose_input, nb_detect_resolution, nb_image_resolution,
|
|
|
|
| 84 |
with gr.Row():
|
| 85 |
btn_generate = gr.Button("GENERATE", variant="primary")
|
| 86 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
btn_generate.click(fn=self.musepose_infer.infer_musepose,
|
| 88 |
inputs=[img_musepose_input, vid_pose_input, weight_dtype, nb_width, nb_height,
|
| 89 |
nb_video_frame_length, nb_video_slice_frame_length,
|