Spaces:

wushuang98
/

test

Runtime error

wushuang98 commited on Jun 2

Commit

84bd0e1

verified ·

1 Parent(s): 658dc18

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -59,11 +59,11 @@ pipe.to("cuda")
 rmbg = BiRefNet(device="cuda")
-@spaces.GPU
 def run_rembg(image: Image.Image):
     return rmbg.run(image)
-@spaces.GPU
 def image2mesh(
     image: Any,
     resolution: str = '1024',

 rmbg = BiRefNet(device="cuda")
+@spaces.GPU(duration=600)
 def run_rembg(image: Image.Image):
     return rmbg.run(image)
+@spaces.GPU(duration=600)
 def image2mesh(
     image: Any,
     resolution: str = '1024',