Update app.py

app.py

@@ -12,8 +12,9 @@ import torch
 from diffusers import DiffusionPipeline
 from typing import Tuple
 
-# Check if GPU is available; fallback to CPU if needed
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# Force CPU usage
+device = torch.device("cpu")
+torch.cuda.is_available = lambda: False
 
 # Setup rules for bad words (ensure the prompts are kid-friendly)
 bad_words = json.loads(os.getenv('BAD_WORDS', '["violence", "blood", "scary", "death", "ghost"]'))
@@ -66,18 +67,17 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
 DESCRIPTION = """## Children's Sticker Generator
 
 Generate fun and playful stickers for children using AI.
-"""
 
-if not torch.cuda.is_available():
-    DESCRIPTION += "\n<p>⚠️Running on CPU, This may be slower.</p>"
+⚠️ Running on CPU. This may be slower.
+"""
 
 MAX_SEED = np.iinfo(np.int32).max
-CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
+CACHE_EXAMPLES = False
 
 # Initialize the DiffusionPipeline
 pipe = DiffusionPipeline.from_pretrained(
-    "SG161222/RealVisXL_V3.0_Turbo",
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    "runwayml/stable-diffusion-v1-5",  # Using a smaller model for CPU
+    torch_dtype=torch.float32,
     use_safetensors=True,
 ).to(device)
 
@@ -142,7 +142,7 @@ def generate(
     # Apply style
     prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device=device).manual_seed(seed)
+    generator = torch.manual_seed(seed)
 
     width, height = size_map.get(size, (512, 512))
 
@@ -157,7 +157,7 @@ def generate(
         "guidance_scale": guidance_scale,
         "num_inference_steps": 20,
         "generator": generator,
-        "num_images_per_prompt": 2,
+        "num_images_per_prompt": 1,  # Reduced to 1 for CPU
         "output_type": "pil",
     }
     
@@ -196,7 +196,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 container=False,
             )
             run_button = gr.Button("Run")
-        result = gr.Gallery(label="Generated Stickers", columns=2, preview=True)
+        result = gr.Gallery(label="Generated Stickers", columns=1, preview=True)
         error_output = gr.Textbox(label="Error", visible=False)
     with gr.Accordion("Advanced options", open=False):
         use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
@@ -236,7 +236,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             minimum=0.1,
             maximum=20.0,
             step=0.1,
-            value=15.7,
+            value=7.5,
         )
 
     gr.Examples(