Update app.py

app.py

@@ -25,6 +25,7 @@ pipe = AutoPipelineForText2Image.from_pretrained(
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler,
   token=auth_token)
+pipe=pipe.to("cuda")
 
 #pipe = AutoPipelineForText2Image.from_pretrained(
 #    model_id, 
@@ -113,7 +114,7 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
 #
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
     conditioning, pooled = compel([prompt, neg_prompt])
-    pipe=pipe.to("cuda")
+
     result = pipe(
         prompt_embeds=conditioning[0:1],
         pooled_prompt_embeds=pooled[0:1],