Update app.py

app.py

@@ -1,7 +1,8 @@
 import os
 import torch
 import gradio as gr
-from PIL import Image
+from PIL import Image, ImageOps
+
 from huggingface_hub import snapshot_download
 from pyramid_dit import PyramidDiTForVideoGeneration
 from diffusers.utils import export_to_video
@@ -61,7 +62,8 @@ def generate_video_from_image(image, prompt, duration, video_guidance_scale):
     temp = int(duration * 2.4)  # Convert seconds to temp value (assuming 24 FPS)
     torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
     
-    image = image.resize((1280, 768))
+    target_size = (1280, 720)
+    image = ImageOps.fit(image, target_size, method=Image.LANCZOS, centering=(0.5, 0.5))
     
     with torch.no_grad(), torch.cuda.amp.autocast(enabled=True, dtype=torch_dtype):
         frames = model.generate_i2v(