dall-e-2 / README.md
BAKULODid's picture
Update README.md
08d74e2
|
raw
history blame
693 Bytes

from transformers import AutoTokenizer, AutoModel

Cargar el tokenizador y el modelo pre-entrenado

tokenizer = AutoTokenizer.from_pretrained("image-tokenizers/dall-e-2") model = AutoModel.from_pretrained("image-generation/dall-e-2")

texto previo y objetos de imagen para alimentar al modelo

text = "Crea una imagen de un perro en una bicicleta" image = "https://miweb.com/imagenes/perro-bicicleta.png"

#Encodear el texto y la imagen inputs = tokenizer(text, image, return_tensors='pt')

#Generar la imagen output = model.generate(**inputs)

#Decodificar la imagen generada decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)

#Imprimir la imagen print(decoded_output)