| import fn | |
| import gradio as gr | |
| with gr.Blocks() as demo: | |
| prompt = gr.Textbox(label='prompt') | |
| negative_prompt = gr.Textbox(label='negative_prompt') | |
| model = gr.Textbox(label='model') | |
| guidance_scale = gr.Textbox(value=5.0, label='guidance_scale') | |
| steps = gr.Textbox(value=20, label='steps') | |
| seed = gr.Textbox(value=-1, label='seed') | |
| run = gr.Button() | |
| dst_image = gr.Image(label="Result", interactive=False) | |
| run.click( | |
| fn=fn.run, | |
| inputs=[prompt, negative_prompt, model, guidance_scale, steps, seed], | |
| outputs=[dst_image], | |
| ) | |
| if __name__ == '__main__': | |
| demo.launch() | |