Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,13 +1,18 @@
|
|
|
|
1 |
import gradio as gr
|
|
|
2 |
from rembg import remove
|
|
|
3 |
from PIL import Image
|
|
|
4 |
import numpy as np
|
|
|
5 |
import io
|
6 |
|
7 |
def remove_bg(image, format, threshold):
|
8 |
input_image = Image.fromarray(np.array(image))
|
9 |
|
10 |
-
#
|
11 |
output_image = remove(input_image)
|
12 |
|
13 |
# Convert to desired format
|
@@ -20,6 +25,13 @@ def remove_bg(image, format, threshold):
|
|
20 |
buffer.seek(0)
|
21 |
return buffer
|
22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
iface = gr.Interface(
|
24 |
fn=remove_bg,
|
25 |
inputs=[
|
|
|
1 |
+
print("import gradio as gr")
|
2 |
import gradio as gr
|
3 |
+
print("from rembg import remove")
|
4 |
from rembg import remove
|
5 |
+
print("from PIL import Image")
|
6 |
from PIL import Image
|
7 |
+
print("import numpy as np")
|
8 |
import numpy as np
|
9 |
+
print("import io")
|
10 |
import io
|
11 |
|
12 |
def remove_bg(image, format, threshold):
|
13 |
input_image = Image.fromarray(np.array(image))
|
14 |
|
15 |
+
# Remove background with threshold adjustment (if applicable)
|
16 |
output_image = remove(input_image)
|
17 |
|
18 |
# Convert to desired format
|
|
|
25 |
buffer.seek(0)
|
26 |
return buffer
|
27 |
|
28 |
+
def convert_pil_to_np(image):
|
29 |
+
return np.array(image)
|
30 |
+
|
31 |
+
def convert_np_to_pil(image):
|
32 |
+
return Image.fromarray(image)
|
33 |
+
|
34 |
+
print("Initialize Interface")
|
35 |
iface = gr.Interface(
|
36 |
fn=remove_bg,
|
37 |
inputs=[
|