Update app.py

app.py

@@ -222,7 +222,7 @@ for _ in range(num_processes):
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Cargar modelos
-text_to_image_pipeline = get_model_or_download("stabilityai/stable-diffusion-3-medium", "text_to_image_model", StableDiffusionPipeline.from_pretrained)
+text_to_image_pipeline = get_model_or_download("stabilityai/stable-diffusion-2", "text_to_image_model", StableDiffusionPipeline.from_pretrained)
 img2img_pipeline = get_model_or_download("CompVis/stable-diffusion-v1-4", "img2img_model", StableDiffusionImg2ImgPipeline.from_pretrained)
 flux_pipeline = get_model_or_download("black-forest-labs/FLUX.1-schnell", "flux_model", FluxPipeline.from_pretrained)
 text_gen_pipeline = transformers_pipeline("text-generation", model="bigcode/starcoder", tokenizer="bigcode/starcoder", device=0)