Spaces:
Sleeping
Sleeping
File size: 2,862 Bytes
771872b 0efebd7 351a99e 771872b 6a1d099 0efebd7 6a1d099 0efebd7 6a1d099 0efebd7 6a1d099 0efebd7 6a1d099 3f285ec 6a1d099 351a99e 0efebd7 771872b 0efebd7 6a1d099 0efebd7 771872b 6a1d099 771872b 6a1d099 771872b d778c10 6a1d099 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
import gradio as gr
from dotenv import load_dotenv
from transformers import pipeline
import urllib.request
from PIL import Image
# Load environment variables from the .env file de forma local
load_dotenv()
import base64
with open("Iso_Logotipo_Ceibal.png", "rb") as image_file:
encoded_image = base64.b64encode(image_file.read()).decode()
import os
import openai
openai.api_key = os.environ["OPENAI_API_KEY"]
es_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-es-en")
def translate_text(text):
text = es_en_translator(text)[0].get("translation_text") # type: ignore
return text
def get_image(text: str, translate: bool):
print(text)
if translate:
text = translate_text(text)
response = openai.Image.create(
prompt=text, n=1, size="512x512", model="dall-e-2"
) # ,response_format="b64_json"
print(response)
urllib.request.urlretrieve(response["data"][0]["url"], "img.png") # type: ignore
img = Image.open("img.png")
return img
# return response["data"][0]["url"] # type: ignore
with gr.Blocks() as demo:
gr.Markdown(
"""
<center>
<h1>
Uso de AI para la generación de imagenes a partir de texto.
</h1>
<img src='data:image/jpg;base64,{}' width=200px>
<h2>
Con este espacio podrás generar imagenes a partir de texto. Utiliza el modelo DALL-E de OpenAI.
</h2>
<h2>
Obtendrás mejores resultados cuanto más larga sea la descripción.
</h2>
</center>
""".format(
encoded_image
)
)
with gr.Row():
with gr.Column():
with gr.Row():
gr.Markdown(
"El modelo funciona mejor con entradas en inglés, podés elegir traducir al inglés con este check:"
)
check = gr.Checkbox(label="Traducir al inglés")
with gr.Row():
gr.Markdown("Primero debes ingresar el texto para generar la imagen:")
with gr.Row():
with gr.Column(scale=4):
prompt = gr.Textbox(
label="Texo base para generar la imagen"
) # Give prompt some real estate
with gr.Column(scale=1, min_width=50):
btn = gr.Button("Generar") # Submit button side by side!
with gr.Column():
output = gr.Image(
label="Resultado", height=512, width=512
) # Move the output up too
# examples = gr.Examples(
# inputs=[prompt]
# examples=[["Un perro en el parque", "low quality"]],
# )
btn.click(
fn=get_image,
inputs=[prompt, check],
outputs=[output],
) # steps,guidance,width,height]
demo.queue(concurrency_count=4)
demo.launch(debug=True)
|