not-lain commited on
Commit
2ca2e3d
·
1 Parent(s): 6ba9354

update api

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -95,19 +95,25 @@ def rmbg(image, url):
95
 
96
  @spaces.GPU
97
  def main(*args, progress=gr.Progress(track_tqdm=True)):
98
- if len(args) == 2:
 
 
99
  return rmbg(*args)
100
- else:
101
  return inpaint(*args)
102
 
103
 
104
  rmbg_tab = gr.Interface(
105
- fn=main, inputs=["image", "text"], outputs=["image"], api_name="rmbg"
 
 
 
106
  )
107
 
108
  outpaint_tab = gr.Interface(
109
  fn=main,
110
  inputs=[
 
111
  "image",
112
  gr.Number(label="padding top"),
113
  gr.Number(label="padding bottom"),
 
95
 
96
  @spaces.GPU
97
  def main(*args, progress=gr.Progress(track_tqdm=True)):
98
+ api_num = args[0]
99
+ args = args[1:]
100
+ if api_num == 1:
101
  return rmbg(*args)
102
+ elif api_num == 2:
103
  return inpaint(*args)
104
 
105
 
106
  rmbg_tab = gr.Interface(
107
+ fn=main,
108
+ inputs=[gr.Number(1, visible=False), "image", "text"],
109
+ outputs=["image"],
110
+ api_name="rmbg",
111
  )
112
 
113
  outpaint_tab = gr.Interface(
114
  fn=main,
115
  inputs=[
116
+ gr.Number(2, visible=False),
117
  "image",
118
  gr.Number(label="padding top"),
119
  gr.Number(label="padding bottom"),