Spaces:
Running
Running
File size: 1,029 Bytes
599a9f1 b52351a 599a9f1 ec62272 b03870d a0c63d9 b03870d b267289 599a9f1 ee74194 a82b385 ee74194 e993e59 ee74194 716d4e9 6f001b5 ee74194 b52351a 5801106 |
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 |
from PIL import Image
import requests
from openai import OpenAI
import gradio as gr
def get_image(prompt,textbox):
prompt = prompt
client = OpenAI(api_key=textbox)
response = client.images.generate(
model = "dall-e-3",
prompt = prompt,
size = "1024x1024",
n = 1
)
image_url = response.data[0].url
data = requests.get(image_url).content
f = open('img.jpg','wb')
f.write(data)
f.close()
img = Image.open('img.jpg')
return img
title = "Please Provide API-Key and Enter your Prompt"
description = """
<img src = "https://upload.wikimedia.org/wikipedia/commons/4/4d/OpenAI_Logo.svg" width=300px>
# A model that can generate images given a natural language prompt
"""
gr.Interface(
get_image,
[
gr.Textbox(label="Enter your Prompt",),
gr.Textbox(type = 'password',label="Enter your API-Key", placeholder="API-Key", lines=1)
],
outputs= gr.Image(type='pil'),
title=title,
description=description,
).launch()
|