Update app.py

app.py

@@ -6,7 +6,6 @@ from musepose_inference import MusePoseInference
 from pose_align import PoseAlignmentInference
 from downloading_weights import download_models
 
-
 class App:
     def __init__(self, args):
         self.args = args
@@ -26,39 +25,32 @@ class App:
         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
-
-                # 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)
-
-                    # Rest of the settings remains the same
-
-            # Event handler and button settings remain unchanged
-
-    return demo
-
+    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)
+                        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
+
+                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)
+                        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
+            return demo
 
     @staticmethod
     def header():
@@ -82,7 +74,6 @@ def musepose_demo(self):
             share=self.args.share
         )
 
-
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument('--model_dir', type=str, default=os.path.join("pretrained_weights"), help='Pretrained models directory for MusePose')