hasibzunair commited on
Commit
0af5e1f
β€’
1 Parent(s): 71f116c
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -172,7 +172,7 @@ thumbnail = None # "./pathtothumbnail.png"
172
 
173
  # todos
174
  # train model with background removed then add feature, also add remove_bg in inferene()
175
- # add gr.inputs.Radio(choices=["yes","no"], default="no", label="Remove background from the person image?") in inputs
176
 
177
  gr.Interface(
178
  inference,
@@ -180,8 +180,8 @@ gr.Interface(
180
  gr.inputs.Image(type='file', label="Person Image"),
181
  gr.inputs.Radio(choices=["yes","no"], label="Retrieve original background from the person image?", type='index')],
182
  gr.outputs.Image(type="file", label="Predicted Output"),
183
- examples=[["./sample_images/1/cloth.jpg", "./sample_images/1/person.jpg"],
184
- ["./sample_images/2/cloth.jpg", "./sample_images/2/person.jpg"]],
185
  title=title,
186
  description=description,
187
  article=article,
 
172
 
173
  # todos
174
  # train model with background removed then add feature, also add remove_bg in inferene()
175
+ # add gr.inputs.Radio(choices=["yes","no"], label="Remove background from the person image?", type='index') in inputs
176
 
177
  gr.Interface(
178
  inference,
 
180
  gr.inputs.Image(type='file', label="Person Image"),
181
  gr.inputs.Radio(choices=["yes","no"], label="Retrieve original background from the person image?", type='index')],
182
  gr.outputs.Image(type="file", label="Predicted Output"),
183
+ examples=[["./sample_images/1/cloth.jpg", "./sample_images/1/person.jpg", "yes"],
184
+ ["./sample_images/2/cloth.jpg", "./sample_images/2/person.jpg", "no"]],
185
  title=title,
186
  description=description,
187
  article=article,