Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def resize_to_match(input_image, output_image):
|
|
31 |
torch_img_scaled = F.interpolate(torch_img.unsqueeze(0),mode='trilinear').squeeze(0)
|
32 |
output_image = torchvision.transforms.functional.to_pil_image(torch_img_scaled, mode=None)
|
33 |
return output_image
|
34 |
-
#return output_image.resize(input_image.size, Image.
|
35 |
|
36 |
# Function to generate the mask using Florence SAM Masking API (Replicate)
|
37 |
def generate_mask(image_path, text_prompt="clothing"):
|
|
|
31 |
torch_img_scaled = F.interpolate(torch_img.unsqueeze(0),mode='trilinear').squeeze(0)
|
32 |
output_image = torchvision.transforms.functional.to_pil_image(torch_img_scaled, mode=None)
|
33 |
return output_image
|
34 |
+
#return output_image.resize(input_image.size, Image.BICUBIC) # Use BILINEAR for faster resizing
|
35 |
|
36 |
# Function to generate the mask using Florence SAM Masking API (Replicate)
|
37 |
def generate_mask(image_path, text_prompt="clothing"):
|