Spaces:
Running
on
L40S
Running
on
L40S
Update gradio_app.py
Browse files- gradio_app.py +1 -1
gradio_app.py
CHANGED
@@ -101,7 +101,7 @@ def generate_and_process_3d(prompt: str, seed: int = 42, width: int = 1024, heig
|
|
101 |
rgb_image = generated_image.convert('RGB')
|
102 |
|
103 |
# Remove background and get mask
|
104 |
-
mask = bg_remover.
|
105 |
mask_uint8 = (mask * 255).astype(np.uint8)
|
106 |
|
107 |
# Create RGBA image
|
|
|
101 |
rgb_image = generated_image.convert('RGB')
|
102 |
|
103 |
# Remove background and get mask
|
104 |
+
mask = bg_remover.process(rgb_image)
|
105 |
mask_uint8 = (mask * 255).astype(np.uint8)
|
106 |
|
107 |
# Create RGBA image
|