fedorajuandy's picture
readd queue
740b9f3
""" Module for web application """
import gradio
from inference import generate_image
with gradio.Blocks() as block:
gradio.Markdown("<h1><center>Image Generator</center></h1>")
text = gradio.Textbox(
label = "Text prompt",
max_lines = 1,
)
btn = gradio.Button("Run")
gallery = gradio.Gallery(
label = "Result image",
).style(
# Number of images per category
columns = [3],
height = "auto",
)
btn.click(fn=generate_image, inputs=text, outputs=gallery)
gradio.Markdown("<p style='text-align: center'>2019130032 - Fedora Yoshe Juandy</p>")
# Use 'queue' if the inference's time > 60s
# Use 'share' for more than one user usage
block.queue().launch()