Spaces:
Running
on
Zero
Running
on
Zero
Update app_3.py
Browse files
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
|