Update app.py

app.py

@@ -1,4 +1,4 @@
-# https://huggingface.co/yuntian-deng/latex2im_ss_finetunegptneo
+import spaces
 import gradio as gr
 import numpy as np
 import requests
@@ -28,6 +28,7 @@ notice = "<p><center>Acknowledgment: This demo is powered by GPU resources suppo
 
 
 # setup
+@spaces.GPU
 def setup(device='cuda'):
     device = ("cuda" if torch.cuda.is_available() else "cpu")
     img_pipe = DDPMPipeline.from_pretrained("yuntian-deng/latex2im_ss_finetunegptneo")