Update app.py

app.py

@@ -11,6 +11,7 @@ from torch import autocast
 from safetensors import safe_open
 from compel import Compel, ReturnedEmbeddingsType
 from safetensors.torch import load_file
+import spaces
 
 model_id = "aipicasso/emix-0-4-turbo"
 auth_token=os.environ["ACCESS_TOKEN"]
@@ -54,6 +55,7 @@ def error_str(error, title="Error"):
     return f"""#### {title}
             {error}"""  if error else ""
 
+@spaces.GPU
 def inference(prompt, guidance, steps, seed=0, neg_prompt="", disable_auto_prompt_correction=False):
   global pipe