Ashoka74 commited on
Commit
1f6d622
1 Parent(s): 38ff1c6

Update app_3.py

Browse files
Files changed (1) hide show
  1. app_3.py +3 -3
app_3.py CHANGED
@@ -382,7 +382,7 @@ def infer(
382
  negative_prompt="watermark, ugly, deformed, noisy, blurry, low contrast",
383
  progress=gr.Progress(track_tqdm=True),
384
  ):
385
- logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
386
 
387
  # Convert input to PIL if needed
388
  if isinstance(image, np.ndarray):
@@ -629,7 +629,7 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
629
  def extract_foreground(image):
630
  if image is None:
631
  return None, gr.update(visible=True), gr.update(visible=True)
632
- logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
633
  #result, rgba = run_rmbg(image)
634
  result = run_rmbg(image)
635
  logging.info(f"Result shape: {result.shape}, dtype: {result.dtype}")
@@ -1290,7 +1290,7 @@ with block:
1290
 
1291
  def convert_to_pil(image):
1292
  try:
1293
- logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
1294
  image = image.astype(np.uint8)
1295
  logging.info(f"Converted image shape: {image.shape}, dtype: {image.dtype}")
1296
  return image
 
382
  negative_prompt="watermark, ugly, deformed, noisy, blurry, low contrast",
383
  progress=gr.Progress(track_tqdm=True),
384
  ):
385
+ #logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
386
 
387
  # Convert input to PIL if needed
388
  if isinstance(image, np.ndarray):
 
629
  def extract_foreground(image):
630
  if image is None:
631
  return None, gr.update(visible=True), gr.update(visible=True)
632
+ #logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
633
  #result, rgba = run_rmbg(image)
634
  result = run_rmbg(image)
635
  logging.info(f"Result shape: {result.shape}, dtype: {result.dtype}")
 
1290
 
1291
  def convert_to_pil(image):
1292
  try:
1293
+ #logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
1294
  image = image.astype(np.uint8)
1295
  logging.info(f"Converted image shape: {image.shape}, dtype: {image.dtype}")
1296
  return image