Update app.py

app.py

@@ -9,7 +9,7 @@ def image_generation(prompt):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
     # Load the Stable Diffusion 2.1 pipeline (as you're using DiffusionPipeline now)
-    pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1",
+    pipeline = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo",
                                                  torch_dtype=torch.float16 if device == "cuda" else torch.float32,
                                                  text_encoder_3=None,
                                                  tokenizer_3=None)