Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 # from diffusers import LTXLatentUpsamplePipeline
 from diffusers import LTXConditionPipeline, LTXLatentUpsamplePipeline
 from diffusers.pipelines.ltx.pipeline_ltx_condition import LTXVideoCondition
-from diffusers.utils import export_to_video, load_video
+from diffusers.utils import export_to_video, load_video, load_image
 import numpy as np
 
 
@@ -53,10 +53,11 @@ def generate(prompt,
     downscaled_height, downscaled_width = round_to_nearest_resolution_acceptable_by_vae(downscaled_height, downscaled_width)
 
     print(mode)
-    if mode == "text-to-video" and (video is not None):
+    if mode == "video-to-video" and (video is not None):
         video = load_video(video)[:frames_to_use]
         condition = True
-    elif mode == "image-to-video" and (image is not None):
+    # elif mode == "image-to-video" and (image is not None):
+    elif mode == "image-to-video":
         #video = [image]
         video =  [load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/penguin.png")]
         condition = True