Spaces:

wushuang98
/

test

Runtime error

wushuang98 commited on Jun 2

Commit

6e0c65b

verified ·

1 Parent(s): 74526e8

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(duration=120)
 def run_rembg(image: Image.Image):
     return rmbg.run(image)
-@spaces.GPU(duration=120)
 def generate_mesh(
     image: Any,
     resolution: str = '1024',

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