import gradio as gr
import os
import sys
from pathlib import Path
models = [
"Linaqruf/animagine-xl",
"nerijs/pixel-art-xl",
"stabilityai/stable-diffusion-xl-base-1.0",
]
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),
gr.Interface.load(f"models/{models[2]}",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("""
SDXL Models models for your enjoyment!
The first time you load a model it takes 200 seconds
But after it's loaded each new image takes 20 seconds to generate!
""")
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(scale=100):
magic1=gr.Textbox(label="Your Prompt", lines=4)
#align-items: center !important;
#appearance: none !important;
#border-style: none !important;
#box-shadow: rgba(0, 0, 0, .2) 0 3px 5px -1px,rgba(0, 0, 0, .14) 0 6px 10px 0,rgba(0, 0, 0, .12) 0 1px 18px 0 !important;
#box-sizing: border-box !important;
#cursor: pointer !important;
#display: inline-flex !important;
#fill: currentcolor !important;
#font-family: "Google Sans",Roboto,Arial,sans-serif !important;
#font-size: 14px !important;
#font-weight: 500 !important;
#height: 48px !important;
#justify-content: center !important;
#letter-spacing: .25px !important;
#line-height: normal !important;
#max-width: 100% !important;
#overflow: visible !important;
#padding: 2px 24px !important;
#position: relative !important;
#text-align: center !important;
#text-transform: none !important;
#transition: box-shadow 280ms cubic-bezier(.4, 0, .2, 1),opacity 15ms linear 30ms,transform 270ms cubic-bezier(0, 0, .2, 1) 0ms !important;
#user-select: none !important;
#-webkit-user-select: none !important;
#touch-action: manipulation !important;
#width: auto !important;
#will-change: transform,opacity !important;
#z-index: 0 !important;
gr.HTML("""""")
run=gr.Button("Generate Image")
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)