upload 7

demo_app.py

@@ -101,14 +101,25 @@ def generate(
         generator = torch.Generator('cuda').manual_seed(seed_value)
 
         with torch.amp.autocast_mode.autocast('cuda', dtype=torch.bfloat16), torch.inference_mode(), torch.no_grad():
-            output = pipe(
-                prompt=prompt,
-                height=height,
-                width=width,
-                num_frames=num_frames,
-                num_inference_steps=num_inference_steps,
-                generator=generator,
-            ).frames[0]
+            # Use image input if provided, else use text prompt
+            if image_input:
+                output = pipe(
+                    image=Image.open(image_input).convert("RGB"),
+                    height=height,
+                    width=width,
+                    num_frames=num_frames,
+                    num_inference_steps=num_inference_steps,
+                    generator=generator,
+                ).frames[0]
+            else:
+                output = pipe(
+                    prompt=prompt,
+                    height=height,
+                    width=width,
+                    num_frames=num_frames,
+                    num_inference_steps=num_inference_steps,
+                    generator=generator,
+                ).frames[0]
 
         output_path = "output.mp4"
         export_to_video(output, output_path, fps=fps)