not-lain commited on
Commit
991bda2
·
1 Parent(s): 46ca072

update app

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -93,8 +93,10 @@ def rmbg(image, url):
93
 
94
  @spaces.GPU
95
  def main(*args, **kwargs):
96
- print(args, kwargs)
97
- return None
 
 
98
 
99
 
100
  rmbg_tab = gr.Interface(
 
93
 
94
  @spaces.GPU
95
  def main(*args, **kwargs):
96
+ if len (args) == 2:
97
+ return rmbg(*args)
98
+ else :
99
+ return inpaint(*args, **kwargs)
100
 
101
 
102
  rmbg_tab = gr.Interface(