Update app.py

app.py

@@ -45,21 +45,65 @@ except:
 
 
 try:
-    print("Enabling vae Slicing")
+    print("Enabling VAE Slicing")
     pipe.enable_vae_slicing()
     print("VAE Slicing Enabled")
-except:
-    print("Model VAE Slicing Failed")
+except Exception as e:
+    print("VAE Slicing Failed")
+    print("Error:", str(e))
+
+
+
+
+# Loading function for Image
+from diffusers.utils import load_image
+
+def prepare_vertical_image(pipe, image_path, base_width=384, base_height=672):
+    """
+    Loads and resizes an image for Wan I2V vertical video generation.
 
+    Args:
+        pipe: WanImageToVideoPipeline (already loaded)
+        image_path (str): Path or URL to image
+        base_width (int): Desired width before adjustment
+        base_height (int): Desired height before adjustment
 
+    Returns:
+        resized_image (PIL.Image)
+        final_width (int)
+        final_height (int)
+    """
 
+    # Load image
+    image = load_image(image_path)
 
+    # Ensure compatibility with Wan spatial constraints
+    mod_value = (
+        pipe.vae_scale_factor_spatial *
+        pipe.transformer.config.patch_size[1]
+    )
 
+    final_width = (base_width // mod_value) * mod_value
+    final_height = (base_height // mod_value) * mod_value
 
+    resized_image = image.resize((final_width, final_height))
+
+    return resized_image, final_width, final_height
+
+
+
+# how to use the Image loading
+image, width, height = prepare_vertical_image(
+    pipe,
+    "input.jpg",
+    base_width=384,
+    base_height=672
+)
 
 
 
 
+@spaces.GPU(duration=60)
 def greet(name):
     return "Hello " + name + "!!"