Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 from PIL import Image
 import gradio as gr
 from huggingface_hub import hf_hub_download, snapshot_download
@@ -99,6 +100,7 @@ def load_model(args):
     return t5_model, vq_model, gpt_model, image_size
 
 
+@spaces.GPU
 def infer(cfg_scale, top_k, top_p, temperature, prompt, seed):
     prompts = [prompt for _ in range(4)]
     caption_embs, emb_masks = t5_model.get_text_embeddings(prompts)