|
import gradio as gr |
|
from modules.default_pipeline import process |
|
|
|
|
|
def generate_clicked(positive_prompt): |
|
return process(positive_prompt=positive_prompt, negative_prompt='bad, ugly') |
|
|
|
|
|
block = gr.Blocks().queue() |
|
with block: |
|
with gr.Column(): |
|
prompt = gr.Textbox(label="Prompt", value='a handsome man in forest') |
|
run_button = gr.Button(label="Run") |
|
result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", height=1280) |
|
run_button.click(fn=generate_clicked, inputs=[prompt], outputs=[result_gallery]) |
|
|
|
|
|
block.launch() |
|
|