Update app.py

app.py

@@ -6,13 +6,13 @@ import torch
 def load_image_generator():
     try:
         model = StableDiffusionXLImg2ImgPipeline.from_pretrained(
-            "stabilityai/stable-diffusion-xl-base-1.0", 
+            "stabilityai/stable-diffusion-2-1", 
             torch_dtype=torch.float16,
             variant="fp16",
             use_safetensors=True
         )
         # Ensure it runs on CPU
-        model = model.to("cpu")
+        #model = model.to("cpu")
         return model
     except Exception as e:
         print(f"Error loading model: {e}")