Update app.py

app.py

@@ -11,11 +11,13 @@ LORA_PATH = "fofr/sdxl-emoji"
 # --- Load the base pipeline ---
 pipe = StableDiffusionXLPipeline.from_pretrained(
     BASE_MODEL,
-    torch_dtype=torch.float16,
-    variant="fp16",           # Use FP16 variant if available for speed
-    safety_checker=None,      # (Optional) disable safety checker to speed things up
+    torch_dtype=torch.float32,  # Use FP32 for CPU
+    variant="fp16",            # You may also need to adjust this if not using GPU
+    safety_checker=None,
 )
-pipe.to("cuda")
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe.to(device)
 
 # --- Enable fast attention if available ---
 try: