Update app.py

app.py

@@ -16,8 +16,8 @@ DESCRIPTION = """ # Instant Image
         ### <span style='color: red;'>You may change the steps from 4 to 8, if you didn't get satisfied results.
         ### First Image processing takes time then images generate faster.
         """
-if not torch.cuda.is_available():
-    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
+#if not torch.cuda.is_available():
+    #DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
@@ -78,7 +78,7 @@ pipe = PixArtAlphaPipeline.from_pretrained(
         "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
         torch_dtype=torch.float16,
         use_safetensors=True,
-).to("cuda:0")
+).to(device)
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])