Update app.py

app.py

@@ -11,7 +11,7 @@ from torch import autocast
 from safetensors import safe_open
 from compel import Compel, ReturnedEmbeddingsType
 from safetensors.torch import load_file
-#import spaces
+import spaces
 
 model_id = "aipicasso/emix-0-4-turbo"
 auth_token=os.environ["ACCESS_TOKEN"]
@@ -55,7 +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
     
@@ -110,7 +110,7 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
             
     return prompt,neg_prompt
 
-#@spaces.GPU
+#
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
     conditioning, pooled = compel([prompt, neg_prompt])