dhanishetty's picture
Update app.py
599a9f1 verified
raw
history blame
610 Bytes
from PIL import Image
import requests
from openai import OpenAI
import gradio as gr
client = OpenAI(api_key= "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")
def get_image(prompt):
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
gr.Interface(fn=get_image,inputs = "textbox", outputs= gr.Image(type='pil')).launch()