Merge pull request #1 from jhj0517/feature/add-examples

app.py

@@ -22,6 +22,7 @@ class App:
 
     def musepose_demo(self):
         with gr.Blocks() as demo:
+            md_header = self.header()
             with gr.Tabs():
                 with gr.TabItem('Step1: Pose Alignment'):
                     with gr.Row():
@@ -39,9 +40,19 @@ class App:
                                 nb_max_frame = gr.Number(label="Max Frame", value=300, precision=0)
 
                             with gr.Row():
-                                btn_algin_pose = gr.Button("ALIGN POSE", variant="primary")
+                                btn_align_pose = gr.Button("ALIGN POSE", variant="primary")
+                    with gr.Column():
+                        examples = [
+                            [os.path.join("assets", "videos", "dance.mp4"), os.path.join("assets", "images", "ref.png"),
+                             512, 720, 0, 300]]
+                        ex_step1 = gr.Examples(examples=examples,
+                                               inputs=[vid_dance_input, img_input, nb_detect_resolution,
+                                                       nb_image_resolution, nb_align_frame, nb_max_frame],
+                                               outputs=[vid_dance_output, vid_dance_output_demo],
+                                               fn=self.pose_alignment_infer.align_pose,
+                                               cache_examples="lazy")
 
-                btn_algin_pose.click(fn=self.pose_alignment_infer.align_pose,
+                btn_align_pose.click(fn=self.pose_alignment_infer.align_pose,
                                      inputs=[vid_dance_input, img_input, nb_detect_resolution, nb_image_resolution,
                                              nb_align_frame, nb_max_frame],
                                      outputs=[vid_dance_output, vid_dance_output_demo])
@@ -76,15 +87,49 @@ class App:
                             with gr.Row():
                                 btn_generate = gr.Button("GENERATE", variant="primary")
 
+                    with gr.Column():
+                        examples = [
+                            [os.path.join("assets", "images", "ref.png"), os.path.join("assets", "videos", "pose.mp4"),
+                             "fp16", 512, 512, 300, 48, 4, 3.5, 99, 20, -1, 1]]
+                        ex_step2 = gr.Examples(examples=examples,
+                                               inputs=[img_input, vid_pose_input, weight_dtype, nb_width, nb_height,
+                                                       nb_video_frame_length, nb_video_slice_frame_length,
+                                                       nb_video_slice_overlap_frame_number, nb_cfg, nb_seed, nb_steps,
+                                                       nb_fps, nb_skip],
+                                               outputs=[vid_output, vid_output_demo],
+                                               fn=self.musepose_infer.infer_musepose,
+                                               cache_examples="lazy")
+
                 btn_generate.click(fn=self.musepose_infer.infer_musepose,
                                    inputs=[img_input, vid_pose_input, weight_dtype, nb_width, nb_height,
-                                           nb_video_frame_length,
-                                           nb_video_slice_frame_length, nb_video_slice_overlap_frame_number, nb_cfg,
-                                           nb_seed,
-                                           nb_steps, nb_fps, nb_skip],
+                                           nb_video_frame_length, nb_video_slice_frame_length,
+                                           nb_video_slice_overlap_frame_number, nb_cfg, nb_seed, nb_steps, nb_fps,
+                                           nb_skip],
                                    outputs=[vid_output, vid_output_demo])
         return demo
 
+    @staticmethod
+    def header():
+        header = gr.HTML(
+            """
+            <style>
+            p, li {
+                font-size: 16px;
+            }
+            </style>
+
+            <h2>Gradio demo for <a href="https://github.com/TMElyralab/MusePose">MusePose</a></h2>
+
+            <p>Demo list you can try in other environment:</p>
+            <ul>
+                <li><a href="https://github.com/jhj0517/MusePose-WebUI"><strong>MusePose WebUI</strong></a> (This repository, you can try in local)</li>
+                <li><a href="https://github.com/jhj0517/stable-diffusion-webui-MusePose.git"><strong>stable-diffusion-webui-MusePose</strong></a> (SD WebUI extension)</li>
+                <li><a href="https://github.com/TMElyralab/Comfyui-MusePose"><strong>Comfyui-MusePose</strong></a> (ComfyUI custom node)</li>
+            </ul>
+            """
+        )
+        return header
+
     def launch(self):
         demo = self.musepose_demo()
         demo.queue().launch()