Ashoka74 commited on
Commit
d2f41dd
1 Parent(s): 146a097

Update app_3.py

Browse files
Files changed (1) hide show
  1. app_3.py +1 -1
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 = process_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)
 
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)