Imadsarvm commited on
Commit
169befd
·
verified ·
1 Parent(s): 73c2ebf

test update for png

Browse files
Files changed (1) hide show
  1. app.py +36 -1
app.py CHANGED
@@ -170,4 +170,39 @@ interface = gr.Interface(
170
  article=article,
171
  allow_flagging='never',
172
  cache_examples=False,
173
- ).queue().launch(show_error=True, share = True)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
170
  article=article,
171
  allow_flagging='never',
172
  cache_examples=False,
173
+ ).queue().launch(show_error=True, share = True)
174
+
175
+ import gradio as gr
176
+ from rembg import remove
177
+
178
+ title = "AI Background Remover"
179
+ description = "To use it, simply upload your image and wait...."
180
+ examples = [['examples/1.jpg', 200, 0.5, 0.7],
181
+ ['examples/2.jpg', 200, 0.5, 0.6],
182
+ ['examples/3.jpg', 200, 0.5, 0.8],
183
+ ['examples/4.jpg', 200, 0.5, 0.8]]
184
+
185
+
186
+ def remove_bg(image):
187
+ res = remove(image)
188
+ return res
189
+
190
+
191
+ inputs = [
192
+ gr.Image(label="Input Image"),
193
+ ]
194
+
195
+ output = gr.Image(label="Output Image")
196
+
197
+ demo = gr.Interface(
198
+ fn=remove_bg,
199
+ title=title,
200
+ description=description,
201
+ examples=examples,
202
+ inputs=inputs,
203
+ outputs=output,
204
+ cache_examples=False,
205
+ allow_flagging="never"
206
+ )
207
+
208
+ demo.launch()