blocks_joined / run.py
freddyaboulton's picture
Upload folder using huggingface_hub
a3871e2 verified
from time import sleep
import gradio as gr
from gradio.media import get_image
# get_image() returns file paths to sample media included with Gradio
cheetah = get_image("cheetah1.jpg")
def img(text):
sleep(3)
return [
cheetah,
cheetah,
cheetah,
cheetah,
cheetah,
cheetah,
cheetah,
cheetah,
cheetah,
]
with gr.Blocks() as demo:
gr.Markdown("<h1><center>DALL路E mini</center></h1>")
gr.Markdown(
"DALL路E mini is an AI model that generates images from any prompt you give!"
)
with gr.Group():
with gr.Row(equal_height=True):
text = gr.Textbox(
label="Enter your prompt",
max_lines=1,
container=False,
)
btn = gr.Button("Run", scale=0)
gallery = gr.Gallery(
label="Generated images",
show_label=False,
columns=1,
height="auto",
)
btn.click(img, inputs=text, outputs=gallery)
if __name__ == "__main__":
demo.launch(css=".container { max-width: 800px; margin: auto; }")
# margin = (TOP, RIGHT, BOTTOM, LEFT)
# rounded = (TOPLEFT, TOPRIGHT, BOTTOMRIGHT, BOTTOMLEFT)