Update pages/textimage.py

pages/textimage.py

@@ -9,17 +9,27 @@ from diffusers import PixArtAlphaPipeline
 # Check for CUDA availability
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
+# # Load the PixArtAlphaPipeline
+# if torch.cuda.is_available():
+#     pipe = PixArtAlphaPipeline.from_pretrained(
+#         "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
+#         torch_dtype=torch.float16,
+#         use_safetensors=True,
+#     )
+#     pipe.to(device)
+#     st.write("Model loaded successfully!")
+# else:
+    # st.error("This demo requires GPU support, which is not available on this system.")
+
 # Load the PixArtAlphaPipeline
-if torch.cuda.is_available():
-    pipe = PixArtAlphaPipeline.from_pretrained(
-        "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
-        torch_dtype=torch.float16,
-        use_safetensors=True,
-    )
-    pipe.to(device)
-    st.write("Model loaded successfully!")
-else:
-    st.error("This demo requires GPU support, which is not available on this system.")
+
+pipe = PixArtAlphaPipeline.from_pretrained(
+    "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
+    torch_dtype=torch.float16,
+    use_safetensors=True,
+)
+pipe.to(device)
+st.write("Model loaded successfully!")
 
 # Constants
 MAX_SEED = np.iinfo(np.int32).max