Spaces:

OmPrakashSingh1704
/

ADVERTISE

Running

OmPrakashSingh1704 commited on Sep 19, 2024

Commit

6b1b6cf

1 Parent(s): 9ee219e

.

Files changed (2) hide show

options/Banner_Model/Image2Image.py CHANGED Viewed

@@ -99,6 +99,7 @@ def I2I(
         num_inference_steps=num_inference_steps_slider
     ).images[0]
     print('INFERENCE DONE')
     return result
 def remove_background(image: Image.Image, threshold: int = 50) -> Image.Image:

         num_inference_steps=num_inference_steps_slider
     ).images[0]
     print('INFERENCE DONE')
+    imageio.imwrite("output_image.png", result)
     return result
 def remove_background(image: Image.Image, threshold: int = 50) -> Image.Image:

options/__pycache__/Banner.cpython-310.pyc CHANGED Viewed

Binary files a/options/__pycache__/Banner.cpython-310.pyc and b/options/__pycache__/Banner.cpython-310.pyc differ