fix: mask image showing
Browse files- 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 |
|