Spaces:
Runtime error
Runtime error
Update app_merged.py
Browse files- app_merged.py +2 -1
app_merged.py
CHANGED
@@ -12,6 +12,7 @@ import spaces
|
|
12 |
import argparse
|
13 |
import random
|
14 |
|
|
|
15 |
import os
|
16 |
import math
|
17 |
import gradio as gr
|
@@ -925,7 +926,7 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
|
|
925 |
def extract_foreground(image):
|
926 |
if image is None:
|
927 |
return None, gr.update(visible=True), gr.update(visible=True)
|
928 |
-
clear_memory()
|
929 |
#logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
|
930 |
#result, rgba = run_rmbg(image)
|
931 |
result = run_rmbg(image)
|
|
|
12 |
import argparse
|
13 |
import random
|
14 |
|
15 |
+
import io
|
16 |
import os
|
17 |
import math
|
18 |
import gradio as gr
|
|
|
926 |
def extract_foreground(image):
|
927 |
if image is None:
|
928 |
return None, gr.update(visible=True), gr.update(visible=True)
|
929 |
+
# clear_memory()
|
930 |
#logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
|
931 |
#result, rgba = run_rmbg(image)
|
932 |
result = run_rmbg(image)
|