feat: optimize image generator for CPU performance

- Switch to RunwayML SD v1.5 model
- Optimize CPU settings and memory usage
- Simplify user interface
- Reduce generation time to 1-2 minutes
- Fix CPU offloading issues

app.py

@@ -5,6 +5,9 @@ import gc
 
 @st.cache_resource
 def load_model():
+    # Explicitly disable CUDA
+    torch.cuda.is_available = lambda : False
+    
     # Use RunwayML's smaller model
     model_id = "runwayml/stable-diffusion-v1-5"
     pipe = StableDiffusionPipeline.from_pretrained(
@@ -13,13 +16,16 @@ def load_model():
         safety_checker=None,
         requires_safety_checker=False,
         use_safetensors=True
-    ).to("cpu")
+    )
     
-    # Optimized settings
+    # Basic CPU optimizations
+    pipe = pipe.to("cpu")
     pipe.enable_attention_slicing(slice_size=1)
-    pipe.enable_vae_slicing()
-    pipe.enable_sequential_cpu_offload()
-    torch.set_num_threads(6)  # Optimal thread count for most systems
+    pipe.enable_vae_tiling()
+    torch.set_num_threads(4)
+    
+    # Clear memory
+    gc.collect()
     
     return pipe
 
@@ -36,17 +42,19 @@ prompt = st.text_area(
 )
 
 if st.button("Generate Image"):
-    with st.spinner("Creating your image... (about 1 minute)"):
+    with st.spinner("Creating your image... (1-2 minutes)"):
         try:
-            # Pre-optimized settings
-            image = pipeline(
-                prompt=prompt,
-                negative_prompt="ugly, blurry, low quality, distorted, disfigured",
-                num_inference_steps=20,
-                guidance_scale=7.0,
-                width=320,
-                height=320,
-            ).images[0]
+            gc.collect()
+            
+            with torch.inference_mode():
+                image = pipeline(
+                    prompt=prompt,
+                    negative_prompt="ugly, blurry, low quality, distorted, disfigured",
+                    num_inference_steps=15,  # Reduced for speed
+                    guidance_scale=7.0,
+                    width=256,  # Reduced for memory
+                    height=256,  # Reduced for memory
+                ).images[0]
             
             # Display and download
             st.image(image, caption=prompt, use_column_width=True)