|
from time import sleep |
|
import gradio as gr |
|
import os |
|
|
|
cheetah = os.path.join(os.path.dirname(__file__), "files/cheetah1.jpg") |
|
|
|
|
|
def img(text): |
|
sleep(3) |
|
return [ |
|
cheetah, |
|
cheetah, |
|
cheetah, |
|
cheetah, |
|
cheetah, |
|
cheetah, |
|
cheetah, |
|
cheetah, |
|
cheetah, |
|
] |
|
|
|
|
|
with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") 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.Box(): |
|
with gr.Row().style(mobile_collapse=False, equal_height=True): |
|
|
|
text = gr.Textbox( |
|
label="Enter your prompt", show_label=False, max_lines=1 |
|
).style( |
|
border=(True, False, True, True), |
|
rounded=(True, False, False, True), |
|
container=False, |
|
) |
|
btn = gr.Button("Run").style( |
|
margin=False, |
|
rounded=(False, True, True, False), |
|
) |
|
gallery = gr.Gallery(label="Generated images", show_label=False).style( |
|
grid=( |
|
1, |
|
3, |
|
), |
|
height="auto", |
|
) |
|
btn.click(img, inputs=text, outputs=gallery) |
|
|
|
|
|
if __name__ == "__main__": |
|
demo.launch() |
|
|
|
|
|
|
|
|
|
|
