Spaces:

TencentARC
/

CustomNet

Running on Zero

jiangyzy commited on Apr 6

Commit

08f3b6b

•

1 Parent(s): 669fdc5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -185,7 +185,7 @@ def load_example(input_image, x0, y0, x1, y1, polar, azimuth, prompt):
     # print(type(polar))
     return input_image, x0, y0, x1, y1, polar, azimuth, prompt
-# @spaces.GPU
 @torch.no_grad()
 def main(args):
     # load model

     # print(type(polar))
     return input_image, x0, y0, x1, y1, polar, azimuth, prompt
+@spaces.GPU
 @torch.no_grad()
 def main(args):
     # load model