File size: 456 Bytes
506e749
 
 
 
 
 
 
 
 
 
 
 
a7951ec
506e749
e7b06d7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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()