Yuanshi commited on
Commit
cec983f
·
verified ·
1 Parent(s): 629e679

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -156,7 +156,7 @@ def infer(
156
 
157
 
158
  if USE_ZERO_GPU:
159
- infer = spaces.GPU(infer, duration=360)
160
 
161
  if __name__ == "__main__":
162
  demo = gradio_interface()
 
156
 
157
 
158
  if USE_ZERO_GPU:
159
+ infer = spaces.GPU(infer)
160
 
161
  if __name__ == "__main__":
162
  demo = gradio_interface()