Spaces:
Paused
Paused
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() | |