Spaces:
Running
on
Zero
Running
on
Zero
Update app_3.py
Browse files
app_3.py
CHANGED
@@ -672,7 +672,7 @@ def extract_foreground(image):
|
|
672 |
#logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
|
673 |
#result, rgba = run_rmbg(image)
|
674 |
result = run_rmbg(image)
|
675 |
-
result =
|
676 |
#logging.info(f"Result shape: {result.shape}, dtype: {result.dtype}")
|
677 |
#logging.info(f"RGBA shape: {rgba.shape}, dtype: {rgba.dtype}")
|
678 |
return result, gr.update(visible=True), gr.update(visible=True)
|
|
|
672 |
#logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
|
673 |
#result, rgba = run_rmbg(image)
|
674 |
result = run_rmbg(image)
|
675 |
+
result = preprocess_image(result)
|
676 |
#logging.info(f"Result shape: {result.shape}, dtype: {result.dtype}")
|
677 |
#logging.info(f"RGBA shape: {rgba.shape}, dtype: {rgba.dtype}")
|
678 |
return result, gr.update(visible=True), gr.update(visible=True)
|