Update app.py

app.py

@@ -128,6 +128,12 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
     seed = math.floor(seed)
     o_width = get_video_dimension(video_path)[0]
     o_height = get_video_dimension(video_path)[1]
+
+    # Prepare dimensions
+    if o_width > 512 :
+        # Calculate the new height while maintaining the aspect ratio
+        o_height = int(o_height / o_width * 512)
+        o_width = 512
     
     # Get FPS of original video input
     target_fps = get_video_dimension(video_path)[2]