Add spaces decorator

app.py

@@ -9,6 +9,7 @@ import os
 import importlib
 from huggingface_hub import hf_hub_download
 from llama_diffusion_model import CustomTransformerModel, CustomTransformerConfig, BidirectionalLlamaAttention, disable_dropout
+import spaces
 
 hf_token = os.getenv("HF_TOKEN")
 
@@ -25,7 +26,7 @@ with open("token_probabilities.json") as f:
     token_probs_dict = json.load(f)
 token_probabilities = np.array([token_probs_dict[str(i)] for i in range(len(token_probs_dict))], dtype=np.float32)
 
-
+@spaces.GPU
 def load_model():
     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_model",