Update app.py

app.py

@@ -17,6 +17,7 @@ from rich import traceback
 from torchvision.transforms.functional import to_tensor
 from transformers import CLIPTokenizer, CLIPTextModel
 from tqdm import tqdm
+import spaces
 
 MODEL_ID = "CompVis/stable-diffusion-v1-4"
 SEED = 1117
@@ -188,6 +189,7 @@ unet.set_attn_processor(
 )
 
 
+@spaces.GPU()
 @torch.inference_mode()
 def inference(
         image_path: str,