Hugging Face's logo

Spaces:
yuntian-deng
/
latex2im
Running on T4

App Files Community
1
yuntian-deng commited on
Commit
5c51951
·
verified ·
1 Parent(s): bc26ef9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ notice = "<p><center>Acknowledgment: This demo is powered by GPU resources suppo
29
 
30
  # setup
31
  @spaces.GPU
32
- def setup(device='cuda'):
33
  device = ("cuda" if torch.cuda.is_available() else "cpu")
34
  img_pipe = DDPMPipeline.from_pretrained("yuntian-deng/latex2im_ss_finetunegptneo")
35
  img_pipe.to(device)
 
29
 
30
  # setup
31
  @spaces.GPU
32
+ def setup():
33
  device = ("cuda" if torch.cuda.is_available() else "cpu")
34
  img_pipe = DDPMPipeline.from_pretrained("yuntian-deng/latex2im_ss_finetunegptneo")
35
  img_pipe.to(device)