Spaces:
Runtime error
Runtime error
File size: 1,791 Bytes
eb60b1e 24fc613 eb60b1e |
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 45 46 47 48 49 50 51 52 53 54 |
#!/usr/bin/env python
# coding: utf-8
import os
import gradio as gr
from backend import get_images_from_backend
block = gr.Blocks(css=".container { max-width: 800px; margin: auto; }")
backend_url = "http://172.28.0.2:8000/generate"
def infer(prompt):
response = get_images_from_backend(prompt, backend_url)
return response["images"]
with block:
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),
margin=False,
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=[3], height="auto"
)
btn.click(infer, inputs=text, outputs=gallery)
gr.Markdown(
"""___
<p style='text-align: center'>
Created by <a href="https://twitter.com/borisdayma" target="_blank">Boris Dayma</a> et al. 2021-2022
<br/>
<a href="https://github.com/borisdayma/dalle-mini" target="_blank">GitHub</a> | <a href="https://wandb.ai/dalle-mini/dalle-mini/reports/DALL-E-mini-Generate-images-from-any-text-prompt--VmlldzoyMDE4NDAy" target="_blank">Project Report</a>
</p>"""
)
block.launch(enable_queue=False)
|