jhj0517
commited on
Commit
•
a6de0d8
1
Parent(s):
a92726a
remove outputs
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,
|