From 1d9b52b108140fa94b4444fe4ee3ead2e3a75e6f Mon Sep 17 00:00:00 2001
From: Yondon Fu <yondon.fu@gmail.com>
Date: Thu, 29 Feb 2024 15:18:13 -0500
Subject: [PATCH] runner: Fixes to image-to-image/video for modal

---
 runner/app/routes/image_to_image.py | 4 ++--
 runner/app/routes/image_to_video.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/runner/app/routes/image_to_image.py b/runner/app/routes/image_to_image.py
index fa4ccfc6..a1df07a9 100644
--- a/runner/app/routes/image_to_image.py
+++ b/runner/app/routes/image_to_image.py
@@ -72,8 +72,8 @@ async def image_to_image(
 
     try:
         images = pipeline(
-            prompt,
-            image,
+            prompt=prompt,
+            image=image,
             strength=strength,
             guidance_scale=guidance_scale,
             negative_prompt=negative_prompt,
diff --git a/runner/app/routes/image_to_video.py b/runner/app/routes/image_to_video.py
index 430335bf..99a98702 100644
--- a/runner/app/routes/image_to_video.py
+++ b/runner/app/routes/image_to_video.py
@@ -60,7 +60,7 @@ async def image_to_video(
 
     try:
         batch_frames = pipeline(
-            Image.open(image.file).convert("RGB"),
+            image=Image.open(image.file).convert("RGB"),
             height=height,
             width=width,
             fps=fps,