Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 from diffusers import AutoPipelineForInpainting, UNet2DConditionModel
 import diffusers
 from share_btn import community_icon_html, loading_icon_html, share_js
+from PIL import Image
 
 pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE").to("cuda")
 
@@ -16,7 +17,7 @@ def read_content(file_path: str) -> str:
     return content
 
 def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, strength=1.0, scheduler="EulerDiscreteScheduler"):
-    original_width, original_height = dict["image"].shape[1:]
+    original_width, original_height = Image.open(dict["image"]).shape[1:]
     if negative_prompt == "":
         negative_prompt = None
     scheduler_class_name = scheduler.split("-")[0]