Spaces:
Running
on
Zero
Running
on
Zero
Update gradio_app.py
Browse files- 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=
|
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")
|