app.py

import gradio as gr

def generate(prompt, n_prompt, modelName):
    return models[modelName].process(prompt, n_prompt)

def create_demo():
    with gr.Blocks() as demo:
        with gr.Column():
            prompt = gr.Textbox(label='Prompt')
            n_prompt = gr.Textbox( label='Negative Prompt', value= 'low res')
            modelName = gr.Dropdown(choices = list(models.keys()), label = "Model", value=list(models.keys())[0])

            run_button = gr.Button('Generar')

        gr.Markdown("lunarnaut txt2img demo")
        result = gr.Gallery(label='Output', show_label=False, elem_id='gallery').style(columns=1, rows=1, preview=True)

        inputs = [
            prompt,
            n_prompt,
            modelName,
        ]

        prompt.submit(
            fn=generate,
            inputs=inputs,
            outputs=result
        )
        n_prompt.submit(
            fn=generate,
            inputs=inputs,
            outputs=result
        )

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


if __name__ == '__main__':
    from model import Model
    models = {
        "Stable Diffusion v1.5": Model("runwayml/stable-diffusion-v1-5"),
        "Anything v3.0": Model("Linaqruf/anything-v3.0"),
        "Realistic Vision v2.0": Model("SG161222/Realistic_Vision_V2.0"),
              }
    demo = create_demo()
    demo.queue().launch()