jbilcke-hf HF staff commited on
Commit
c9fb81c
·
verified ·
1 Parent(s): daf9fe6

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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.process_image(rgb_image)
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