Spaces:
Running
Running
import gradio as gr | |
import os | |
import sys | |
from pathlib import Path | |
models = [ | |
"Yntec/OpenLexica", | |
"Yntec/MapleSyrup", | |
] | |
current_model = models[0] | |
text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link") | |
models2=[ | |
gr.Interface.load(f"models/{models[0]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[1]}",live=True,preprocess=False), | |
] | |
def text_it1(inputs,text_gen1=text_gen1): | |
go_t1=text_gen1(inputs) | |
return(go_t1) | |
def set_model(current_model): | |
current_model = models[current_model] | |
return gr.update(label=(f"{current_model}")) | |
def send_it1(inputs, model_choice): | |
proc1=models2[model_choice] | |
output1=proc1(inputs) | |
return(output1) | |
css="""""" | |
with gr.Blocks(css=css) as myface: | |
gr.HTML() | |
with gr.Row(): | |
with gr.HTML(""" <title>Minimum Multiplier</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;"> | |
<h1>Fill the Textbox at the top and click Generate Image</h1> | |
<br><h4>The first time you load a model it takes 200 seconds</h4> | |
<br><h4>But after it loads each image takes 20 seconds to generate!</h4> | |
"""): | |
with gr.Row(): | |
with gr.Column(scale=100): | |
magic1=gr.Textbox(lines=4) | |
gr.HTML("""<style> .gr-button { | |
color: white !important; | |
border-color: #000000 !important; | |
background: #006699 !important; | |
}</style>""") | |
run=gr.Button("Generate Image") | |
with gr.Row(): | |
with gr.Column(scale=100): | |
#Model selection dropdown | |
model_name1 = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model, interactive=True) | |
with gr.Row(): | |
with gr.Column(style="width=800px"): | |
output1=gr.Image(label=(f"{current_model}")) | |
with gr.Row(): | |
with gr.Column(scale=50): | |
input_text=gr.Textbox(label="Use this box to extend an idea automagically, by typing some words and clicking Extend Idea",lines=2) | |
use_short=gr.Button("Use Short Prompt") | |
see_prompts=gr.Button("Extend Idea") | |
def short_prompt(inputs): | |
return(inputs) | |
model_name1.change(set_model,inputs=model_name1,outputs=[output1]) | |
run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1]) | |
use_short.click(short_prompt,inputs=[input_text],outputs=magic1) | |
see_prompts.click(text_it1,inputs=[input_text],outputs=magic1) | |
myface.queue(concurrency_count=200) | |
myface.launch(inline=True, show_api=False, max_threads=400) |