Update app.py

app.py

@@ -76,9 +76,10 @@ NUM_IMAGES_PER_PROMPT = 1
 
 pipe = PixArtAlphaPipeline.from_pretrained(
         "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
-        torch_dtype=torch.float16,
+        torch_dtype=torch.float16, 
         use_safetensors=True,
-).to("cuda:0")
+).to("cpu") 
+
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])