File size: 693 Bytes
08d74e2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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)