Update image_generator.py

image_generator.py

@@ -22,7 +22,7 @@ class ImageGenerator:
                     # Using a lightweight model for image generation
                     from diffusers import StableDiffusionPipeline
                     
-                    model_id = "runwayml/stable-diffusion-v1-5"
+                    model_id = "sd-legacy/stable-diffusion-v1-5"
                     
                     # Load with memory optimization settings
                     self.model = StableDiffusionPipeline.from_pretrained(