File size: 811 Bytes
d2790f2 0d07e38 d2790f2 806b780 d2790f2 43098ef 0d07e38 43098ef 0d07e38 d07aad8 43098ef 0d07e38 d07aad8 43098ef f88f6dc ede04cf 4bf3352 |
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 |
import gradio as gr
import openai
import os
openai.api_key = os.environ.get("OPENAI_API_KEY")
# Generate an image from text using DALL-E
def open_ai_txt2img(prompt):
try:
response = openai.Image.create(
prompt=prompt,
n=3,
size="1024x1024",
)
return [img["url"] for img in response["data"]]
except openai.error.OpenAIError as e:
print(e.http_status)
print(e.error)
# Gradio Interface
def generator(prompt):
return open_ai_txt2img(prompt)
with gr.Blocks() as demo:
prompt = gr.Textbox(label="Prompt")
submit = gr.Button(label="Generate")
image1, image2, image3 = gr.Image(), gr.Image(), gr.Image()
submit.click(generator, inputs=[prompt], outputs=[image1, image2, image3], api_name="mmd")
demo.launch()
|