| library_name: diffusers | |
| pipeline_tag: text-to-image | |
| tags: | |
| - text-to-image | |
| - image-generation | |
| license: mit | |
| datasets: | |
| - your-dataset-name | |
| base_model: stabilityai/stable-diffusion-2-1 | |
| thumbnail: "https://your-thumbnail-link.com/thumbnail.png" | |
| # Nombre del Modelo | |
| Este modelo está diseñado para generar imágenes a partir de descripciones de texto utilizando la arquitectura Stable Diffusion. | |
| ## Descripción | |
| Este modelo fue entrenado utilizando el conjunto de datos `your-dataset-name` y está basado en el modelo `stabilityai/stable-diffusion-2-1`. Está optimizado para tareas de generación de imágenes a partir de texto, utilizando la librería `diffusers`. | |
| ## Cómo Usar | |
| Puedes usar este modelo directamente con la Inference API de Hugging Face. Aquí tienes un ejemplo en Python: | |
| ```python | |
| from diffusers import StableDiffusionPipeline | |
| import torch | |
| model_id = "tu-usuario/tu-modelo" | |
| pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16) | |
| pipe = pipe.to("cuda") | |
| prompt = "Una pintura surrealista de un gato volando en el espacio" | |
| image = pipe(prompt).images[0] | |
| image.save("output.png") | |
| ``` | |