ihsanvp commited on
Commit
76f85f6
1 Parent(s): d423351

fix: mask image showing

Browse files
Files changed (1) hide show
  1. models/product.py +0 -1
models/product.py CHANGED
@@ -23,7 +23,6 @@ class ProductBackgroundModifier:
23
  def generate(self, image: Image.Image, prompt: str) -> Image.Image:
24
  image_tensor = self.transform(image).to(self.device)
25
  mask_image = self.segmentation_model.generate(image_tensor)
26
- mask_image.show()
27
  generated_image = self.inpainting_model.generate(image=image, mask_image=mask_image, prompt=prompt)
28
  return generated_image
29
 
 
23
  def generate(self, image: Image.Image, prompt: str) -> Image.Image:
24
  image_tensor = self.transform(image).to(self.device)
25
  mask_image = self.segmentation_model.generate(image_tensor)
 
26
  generated_image = self.inpainting_model.generate(image=image, mask_image=mask_image, prompt=prompt)
27
  return generated_image
28