Update app.py

app.py

@@ -30,7 +30,7 @@ def get_frame_count(filepath):
     
         video.release()
 
-        LIMITS
+        # LIMITS
         if frame_count > 36 :
             frame_count = 36 # limit to 236 frames to avoid cuDNN errors
 
@@ -200,12 +200,13 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("""
             <h1 style="text-align: center;">ControlVideo</h1>
+            <p> Pytorch implementation of "ControlVideo: Training-free Controllable Text-to-Video Generation" </p>
         """)
         with gr.Row():
             with gr.Column():
                 #video_in = gr.Video(source="upload", type="filepath", visible=True)
-                video_path = gr.Video(source="upload", type="filepath", visible=True)
-                prompt = gr.Textbox(label="prompt")
+                video_path = gr.Video(source="upload", type="filepath", visible=True, elem_id="video-in")
+                prompt = gr.Textbox(label="prompt", elem_id="prompt-in")
                 with gr.Column():
                     video_length = gr.Slider(label="Video length", info="How many frames do you want to process ? For demo purpose, max is set to 24", minimum=1, maximum=12, step=1, value=2)
                     with gr.Row():
@@ -214,10 +215,10 @@ with gr.Blocks(css=css) as demo:
                     inference_steps = gr.Slider(label="Inference steps", minimum=25, maximum=50, step=1, value=25)
                 submit_btn = gr.Button("Submit")
             with gr.Column():
-                video_res = gr.Video(label="result")
+                video_res = gr.Video(label="result", elem_id="video-out")
                 status = gr.Textbox(label="result")
                 gr.Examples(
-                    examples=[["Indiana Jones moonwalk in the Jungle", "./examples/moonwalk.mp4", 'depth', 12, 42, 50]],
+                    examples=[["Indiana Jones moonwalk in the Jungle", "./examples/moonwalk.mp4", 'depth', 12, 424242, 50]],
                     fn=run_inference,
                     inputs=[prompt,
                              video_path,