Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ def dilate_mask(mask, kernel_size=5, iterations=5):
|
|
33 |
|
34 |
@spaces.GPU
|
35 |
def remove_obj(image, uploaded_mask, seed):
|
36 |
-
image_pil = image.resize((512, 512), Image.
|
37 |
mask = dilate_mask(uploaded_mask)
|
38 |
seed = int(seed)
|
39 |
latents = torch.randn((1, 4, 64, 64), generator=torch.Generator().manual_seed(seed)).to("cuda")
|
|
|
33 |
|
34 |
@spaces.GPU
|
35 |
def remove_obj(image, uploaded_mask, seed):
|
36 |
+
image_pil = image.resize((512, 512), Image.LANCZOS)
|
37 |
mask = dilate_mask(uploaded_mask)
|
38 |
seed = int(seed)
|
39 |
latents = torch.randn((1, 4, 64, 64), generator=torch.Generator().manual_seed(seed)).to("cuda")
|