nolenfelten commited on
Commit
997f1b0
·
verified ·
1 Parent(s): 3aa0fcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
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
- # Background removal processing
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=[