not-lain commited on
Commit
c74d7e5
·
1 Parent(s): a181ad3

update rmbg

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -134,11 +134,16 @@ def main(*args):
134
 
135
  rmbg_tab = gr.Interface(
136
  fn=main,
137
- inputs=[gr.Number(1, visible=False), "image", "text"],
 
 
 
 
138
  outputs=["image"],
139
  api_name="rmbg",
140
  examples=[[1, "./assets/Inpainting mask.png", ""]],
141
  cache_examples=False,
 
142
  )
143
 
144
  outpaint_tab = gr.Interface(
 
134
 
135
  rmbg_tab = gr.Interface(
136
  fn=main,
137
+ inputs=[
138
+ gr.Number(1, visible=False),
139
+ "image",
140
+ gr.Text("", label="url"),
141
+ ],
142
  outputs=["image"],
143
  api_name="rmbg",
144
  examples=[[1, "./assets/Inpainting mask.png", ""]],
145
  cache_examples=False,
146
+ description="pass an image or a url of an image",
147
  )
148
 
149
  outpaint_tab = gr.Interface(