Update app.py

app.py

@@ -23,9 +23,9 @@ model_id2 = "stabilityai/stable-diffusion-xl-base-1.0"
 
 # Initialize Stable Diffusion pipeline based on device
 if torch.cuda.is_available():
-    pipe = StableDiffusionPipeline.from_pretrained(model_id2, torch_dtype=torch.float16)
+    pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16)
 else:
-    pipe = StableDiffusionPipeline.from_pretrained(model_id2)  # Omit torch_dtype for CPU
+    pipe = StableDiffusionPipeline.from_pretrained(model_id1)  # Omit torch_dtype for CPU
 
 # Move model to the selected device (either GPU or CPU)
 pipe = pipe.to(device)