not-lain commited on
Commit
7b04047
·
verified ·
1 Parent(s): b8ddbe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -161,7 +161,7 @@ def erase(image=None, mask=None):
161
  return simple_lama(image, mask)
162
 
163
 
164
- @spaces.GPU
165
  def main(*args):
166
  api_num = args[0]
167
  args = args[1:]
 
161
  return simple_lama(image, mask)
162
 
163
 
164
+ @spaces.GPU(duration=120)
165
  def main(*args):
166
  api_num = args[0]
167
  args = args[1:]