mcrlc commited on
Commit
15da770
1 Parent(s): c24b6f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -123,7 +123,7 @@ hypar["model"] = ISNetDIS()
123
  net = build_model(hypar, device)
124
 
125
 
126
- def inference(image: Image, color: str):
127
  image_path = image
128
 
129
  image_tensor, orig_size = load_image(image_path, hypar)
@@ -152,8 +152,7 @@ article = "<div><center><img src='https://visitor-badge.glitch.me/badge?page_id=
152
  interface = gr.Interface(
153
  fn=inference,
154
  inputs=[
155
- gr.Image(type='filepath', label="Image"),
156
- gr.ColorPicker(label="Background Color", value="0x00C49A")
157
  ],
158
  outputs=[
159
  gr.Image(label="Full Background"),
 
123
  net = build_model(hypar, device)
124
 
125
 
126
+ def inference(image: Image):
127
  image_path = image
128
 
129
  image_tensor, orig_size = load_image(image_path, hypar)
 
152
  interface = gr.Interface(
153
  fn=inference,
154
  inputs=[
155
+ gr.Image(type='filepath', label="Image")
 
156
  ],
157
  outputs=[
158
  gr.Image(label="Full Background"),