Update app.py

app.py

@@ -41,6 +41,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):
   generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
 
@@ -94,7 +95,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_conditioning, neg_pooled = compel(neg_prompt)