Spaces:
Build error
Build error
File size: 1,161 Bytes
ee103d5 d6752a6 ee103d5 d6752a6 ee103d5 d6752a6 ee103d5 64e5cbd ee103d5 78610a7 ee103d5 |
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 |
import gradio as gr
import requests
imort openai
from PIL import Image
from io import BytesIO
openai.api_key = os.getenv("OPENAI_API_KEY")
# Define the function that generates the image
def generate_image(prompt):
response = requests.post("https://api.openai.com/v1/images/generations", json={
"model": "image-alpha-001",
"prompt": prompt,
"num_images": 4,
"size": "1024x1024",
"response_format": "url"
}, headers={
"Content-Type": "application/json",
"Authorization": "Bearer "
})
response.raise_for_status()
image_url = response.json()["data"][0]["url"]
image = Image.open(BytesIO(requests.get(image_url).content))
return image
iface = gr.Interface(
fn=generate_image,
inputs=gr.inputs.Textbox(label="Enter Prompt Here"),
outputs="image",
examples=[
["a cat sitting on a couch"],
["a robot walking in the park"],
["a tree made of clouds"],
],
title="TalkGPT Image Generation",
description="Use AI to generate images based on a prompt.",
allow_flagging=False,
analytics_enabled=True,
theme="compact"
)
|