Comming Soon

app.py

@@ -4,7 +4,7 @@ import torch
 from model import Model, ModelType
 
 # from app_canny import create_demo as create_demo_canny
-# from app_pose import create_demo as create_demo_pose
+from app_pose import create_demo as create_demo_pose
 from app_text_to_video import create_demo as create_demo_text_to_video
 from app_pix2pix_video import create_demo as create_demo_pix2pix_video
 # from app_canny_db import create_demo as create_demo_canny_db
@@ -44,8 +44,8 @@ with gr.Blocks(css='style.css') as demo:
         # pass
         create_demo_pix2pix_video(model)
     with gr.Tab('Pose Conditional'):
-        pass
-        # create_demo_pose(model)
+        # pass
+        create_demo_pose(model)
     with gr.Tab('Edge Conditional'):
         pass
         # create_demo_canny(model)

app_pose.py

@@ -3,61 +3,9 @@ import os
 
 from model import Model
 
-examples = [
-    ['Motion 1', "A Robot is dancing in Sahara desert"],
-    ['Motion 2', "A Robot is dancing in Sahara desert"],
-    ['Motion 3', "A Robot is dancing in Sahara desert"],
-    ['Motion 4', "A Robot is dancing in Sahara desert"],
-    ['Motion 5', "A Robot is dancing in Sahara desert"],
-]
-
 def create_demo(model: Model):
     with gr.Blocks() as demo:
         with gr.Row():
-            gr.Markdown('## Text and Pose Conditional Video Generation')
-
-        with gr.Row():
-            gr.Markdown('### You must select one pose sequence shown below, or use the examples')
-            with gr.Column():
-                gallery_pose_sequence = gr.Gallery(label="Pose Sequence", value=[('__assets__/poses_skeleton_gifs/dance1.gif', "Motion 1"), ('__assets__/poses_skeleton_gifs/dance2.gif', "Motion 2"), ('__assets__/poses_skeleton_gifs/dance3.gif', "Motion 3"), ('__assets__/poses_skeleton_gifs/dance4.gif', "Motion 4"), ('__assets__/poses_skeleton_gifs/dance5.gif', "Motion 5")]).style(grid=[2], height="auto")
-                input_video_path = gr.Textbox(label="Pose Sequence",visible=False,value="Motion 1")
-                gr.Markdown("## Selection")
-                pose_sequence_selector = gr.Markdown('Pose Sequence: **Motion 1**')
-            with gr.Column():
-                prompt = gr.Textbox(label='Prompt')
-                run_button = gr.Button(label='Run')
-            with gr.Column():
-                # result = gr.Video(label="Generated Video")
-                result = gr.Image(label="Generated Video")
-
-        input_video_path.change(on_video_path_update, None, pose_sequence_selector)
-        gallery_pose_sequence.select(pose_gallery_callback, None, input_video_path)
-        inputs = [
-            input_video_path,
-            #pose_sequence, 
-            prompt,
-        ]
-
-        gr.Examples(examples=examples,
-                    inputs=inputs,
-                    outputs=result,
-                    # cache_examples=os.getenv('SYSTEM') == 'spaces',
-                    fn=model.process_controlnet_pose,
-                    run_on_click=False,
-                    )
-        #fn=process,
-        #)
-
-
-        run_button.click(fn=model.process_controlnet_pose,
-                         inputs=inputs,
-                         outputs=result,)
+            gr.Markdown('## Text and Pose Conditional Video Generation (coming soon)')
 
     return demo
-
-
-def on_video_path_update(evt: gr.EventData):
-    return f'Pose Sequence: **{evt._data}**'
-
-def pose_gallery_callback(evt: gr.SelectData):
-    return f"Motion {evt.index+1}"