Update app.py

app.py

@@ -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
     
@@ -109,12 +109,11 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
             return prompt, neg_prompt
             
     return prompt,neg_prompt
-    
+
+@spaces.GPU
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
     conditioning, pooled = compel([prompt, neg_prompt])
-    conditioning=conditioning.to("cuda")
-    pooled=pooled.to("cuda")
-    
+
     result = pipe(
         prompt_embeds=conditioning[0:1],
         pooled_prompt_embeds=pooled[0:1],