Update app.py

app.py

@@ -46,9 +46,14 @@ def generate(prompt,
     downscaled_height, downscaled_width = round_to_nearest_resolution_acceptable_by_vae(downscaled_height, downscaled_width)
 
     condition = image is not None or video is not None
+    if video:
+        frames_to_use = 21
+        video = load_video(video)[:frames_to_use]
+    else:
+        video = [image]
 
     if condition and (not t2v):
-        condition1 = LTXVideoCondition(video=[image], frame_index=0)
+        condition1 = LTXVideoCondition(video=video, frame_index=0)
         latents = pipe(
             conditions=condition1,
             prompt=prompt,
@@ -164,7 +169,7 @@ with gr.Blocks(css=css, theme=gr.themes.Ocean()) as demo:
           image = gr.Image(label="")
           #prompt = gr.Textbox(label="prompt")
         with gr.Tab("video-to-video"):
-          image = gr.Video(label="")
+          video = gr.Video(label="")
         prompt = gr.Textbox(label="prompt")
       run_button = gr.Button()
     with gr.Column():