dalle-3 / app.py
Kvikontent's picture
Update app.py
a7951ec
raw
history blame
456 Bytes
import gradio as gr
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("dataautogpt3/OpenDalleV1.1")
def generate_image(prompt):
if prompt != '':
image = pipeline.generate(prompt=prompt)
return image
else:
return None
prompt_input = gr.Textbox(label="Enter your prompt here:")
gr.Interface(generate_image, inputs=gr.Textbox(lines=5), outputs=gr.Image(label="Generated Image")).launch()