Update app.py

app.py

@@ -23,45 +23,57 @@ class App:
 
     @staticmethod
     def on_step1_complete(input_img: str, input_pose_vid: str):
-
         return [gr.Image(label="Input Image", value=input_img, type="filepath", scale=5),
                 gr.Video(label="Input Aligned Pose Video", value=input_pose_vid, scale=5)]
 
-def musepose_demo(self):
-    with gr.Blocks() as demo:
-        md_header = self.header()
-        with gr.Tabs():
-            with gr.TabItem('1: Pose Alignment'):
-                with gr.Row():
-                    with gr.Column(scale=3):
-                        img_pose_input = gr.Image(label="Input Image", type="filepath", scale=5)
-                        vid_dance_input = gr.Video(label="Input Dance Video", max_length=10, scale=5)  # Changed max_length to 10
-                    with gr.Column(scale=3):
-                        vid_dance_output = gr.Video(label="Aligned Pose Output", scale=5, interactive=False)
-                        vid_dance_output_demo = gr.Video(label="Aligned Pose Output Demo", scale=5)
-                    # rest of the column setup remains the same
-                    with gr.Column(scale=3):
-                        # column settings remain the same
+    def musepose_demo(self):
+        with gr.Blocks() as demo:
+            md_header = self.header()
+            with gr.Tabs():
+                with gr.TabItem('1: Pose Alignment'):
+                    with gr.Row():
+                        with gr.Column(scale=3):
+                            img_pose_input = gr.Image(label="Input Image", type="filepath", scale=5)
+                            vid_dance_input = gr.Video(label="Input Dance Video", max_length=10, scale=5)  # Changed max_length to 10
+                        with gr.Column(scale=3):
+                            vid_dance_output = gr.Video(label="Aligned Pose Output", scale=5, interactive=False)
+                            vid_dance_output_demo = gr.Video(label="Aligned Pose Output Demo", scale=5)
+                        # Rest of the column setup remains the same
+                        with gr.Column(scale=3):
+                            # column settings remain the same
 
                 # button settings and event handlers remain the same
 
-            with gr.TabItem('2: MusePose Inference'):
-                with gr.Row():
-                    with gr.Column(scale=3):
-                        img_musepose_input = gr.Image(label="Input Image", type="filepath", scale=5)
-                        vid_pose_input = gr.Video(label="Input Aligned Pose Video", max_length=10, scale=5)  # Changed max_length to 10
-                    with gr.Column(scale=3):
-                        vid_output = gr.Video(label="MusePose Output", scale=5)
-                        vid_output_demo = gr.Video(label="MusePose Output Demo", scale=5)
+                with gr.TabItem('2: MusePose Inference'):
+                    with gr.Row():
+                        with gr.Column(scale=3):
+                            img_musepose_input = gr.Image(label="Input Image", type="filepath", scale=5)
+                            vid_pose_input = gr.Video(label="Input Aligned Pose Video", max_length=10, scale=5)  # Changed max_length to 10
+                        with gr.Column(scale=3):
+                            vid_output = gr.Video(label="MusePose Output", scale=5)
+                            vid_output_demo = gr.Video(label="MusePose Output Demo", scale=5)
 
-                    # rest of the settings remains the same
+                        # Rest of the settings remains the same
 
             # Event handler and button settings remain unchanged
 
-    return demo
-
-@staticmethod
+        return demo
 
+    @staticmethod
+    def header():
+        header = gr.HTML(
+            """
+            <h1 style="font-size: 23px;">
+                <a href="https://github.com/jhj0517/MusePose-WebUI" target="_blank">MusePose WebUI</a>
+            </h1>
+            <p style="font-size: 18px;">
+                <strong>Note</strong>: This space now allows video input up to <strong>10 seconds</strong> because ZeroGPU limits the function runtime to 2 minutes. <br>
+                If you want longer video inputs, you have to run it locally. Click the link above and follow the README to try it locally.<br><br>
+                When you have completed the <strong>1: Pose Alignment</strong> process, go to <strong>2: MusePose Inference</strong> and click the "GENERATE" button.
+            </p>
+            """
+        )
+        return header
 
     def launch(self):
         demo = self.musepose_demo()
@@ -79,4 +91,4 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     app = App(args=args)
-    app.launch()
+    app.launch()