Update app.py

app.py

@@ -8,9 +8,9 @@ import time
 device = "cpu"
 print(f"Using device: {device}")
 
-# Load a smaller, CPU-friendly model
-# Using 'Oppenheimer/DALL-E_1.5' - a fine-tuned SD 1.5 model known for good results
-model_id = "Oppenheimer/DALL-E_1.5"
+# Load a smaller, CPU-friendly PUBLIC model
+# Using 'dreamlike-art/dreamlike-diffusion-1.0' - a public fine-tuned model
+model_id = "dreamlike-art/dreamlike-diffusion-1.0"
 
 print("Loading pipeline... This may take a few minutes.")
 try:
@@ -18,7 +18,9 @@ try:
     pipe = StableDiffusionPipeline.from_pretrained(
         model_id,
         torch_dtype=torch.float32, # Critical for CPU
-        use_safetensors=True
+        use_safetensors=True,
+        safety_checker=None,  # Disable safety checker to avoid NSFW filters that can cause issues
+        requires_safety_checker=False  # Newer versions of diffusers need this
     )
     
     # Optional: Use a faster scheduler for quicker generation