File size: 637 Bytes
d15c3ce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import gradio as gr
from inference import generate_images
def infer(prompt):
images = generate_images(prompt, model_path="output/", num_images=1)
return images[0]
with gr.Blocks() as demo:
with gr.Row():
gr.Markdown("## Image Generation Demo")
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label="Prompt", lines=2)
submit = gr.Button("Generate Image")
with gr.Column():
output_image = gr.Image(label="Generated Image")
submit.click(infer, inputs=prompt, outputs=output_image)
if __name__ == "__main__":
demo.launch()
|