Update app.py

app.py

@@ -33,7 +33,7 @@ def dilate_mask(mask, kernel_size=5, iterations=5):
 
 @spaces.GPU
 def remove_obj(image, uploaded_mask, seed):
-    image_pil = image["image"].resize((512, 512), Image.ANTIALIAS)
+    image_pil = image.resize((512, 512), Image.ANTIALIAS)
     mask = dilate_mask(uploaded_mask)
     seed = int(seed)
     latents = torch.randn((1, 4, 64, 64), generator=torch.Generator().manual_seed(seed)).to("cuda")