Update app.py

app.py

@@ -5,7 +5,7 @@ from diffusers import StableDiffusion3Pipeline
 
 def image_generation(prompt):
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",
+    pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers",
                                                         torch_dtype=torch.float16 if device == "cuda" else torch.float32,
                                                         text_encoder_3 =None,
                                                         tokenizer_3 =None)