ouclxy commited on
Commit
2cc7422
·
verified ·
1 Parent(s): cab03fb

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -403,7 +403,7 @@ with open("imgs/background.png", "rb") as f:
403
  _b64_bg = base64.b64encode(f.read()).decode()
404
 
405
 
406
- @spaces.GPU(duration=180)
407
  def inference(id_image, hair_image):
408
  # ZeroGPU: 强制使用 'cuda' 设备(ZeroGPU 下 torch.cuda.is_available 可能为 False)。
409
  device = torch.device("cuda")
 
403
  _b64_bg = base64.b64encode(f.read()).decode()
404
 
405
 
406
+ @spaces.GPU(duration=300)
407
  def inference(id_image, hair_image):
408
  # ZeroGPU: 强制使用 'cuda' 设备(ZeroGPU 下 torch.cuda.is_available 可能为 False)。
409
  device = torch.device("cuda")