ciover2024 commited on
Commit
0088107
·
verified ·
1 Parent(s): 5f34151

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.BILINEAR) # 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"):
 
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"):