Update app.py

app.py

@@ -10,6 +10,7 @@ from skyreelsinfer import TaskType
 from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image
+from PIL import Image
 
 #predictor = None
 #task_type = None
@@ -49,7 +50,7 @@ def generate_video(prompt, seed, image=None):
         "cfg_for": False,
     }
     assert image is not None, "please input image"
-    kwargs["image"] = load_image(image=image)
+    kwargs["image"] = load_image(image=image).resize((320,320), Image.LANCZOS)
     #global predictor
     output = predictor.inference(kwargs)
     save_dir = f"./"