File size: 1,120 Bytes
2df9553
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
869b788
2df9553
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import gradio as gr

def create_demo(process):
    with gr.Blocks() as demo:
        with gr.Column():
            prompt = gr.Textbox(label='Prompt')
            n_prompt = gr.Textbox(
                label='Negative Prompt',
                value=
                'low quality, ugly, disfigured, deformed'
            )

            run_button = gr.Button('Run')
        result = gr.Gallery(label='Output',
                            show_label=False,
                            elem_id='gallery').style(columns=1, rows=1, preview=True)        

        inputs = [
            prompt,
            n_prompt
        ]
        prompt.submit(
            fn=process,
            inputs=inputs,
            outputs=result
        )
        n_prompt.submit(
            fn=process,
            inputs=inputs,
            outputs=result
        )

        run_button.click(
            fn=process,
            inputs=inputs,
            outputs=result
        )
    return demo


if __name__ == '__main__':
    from model import Model
    model = Model()    
    demo = create_demo(model.process)
    demo.queue().launch()