Approximetal commited on
Commit
484e4a0
·
verified ·
1 Parent(s): c232233

Update inference_gradio.py

Browse files
Files changed (1) hide show
  1. inference_gradio.py +1 -1
inference_gradio.py CHANGED
@@ -235,7 +235,7 @@ def get_available_projects():
235
  print("project_list:", project_list)
236
  return project_list
237
 
238
- @spaces.GPU
239
  @torch.no_grad()
240
  @torch.inference_mode()
241
  def infer(
 
235
  print("project_list:", project_list)
236
  return project_list
237
 
238
+ @spaces.GPU(duration=240)
239
  @torch.no_grad()
240
  @torch.inference_mode()
241
  def infer(