Spaces:
Runtime error
Runtime error
File size: 5,848 Bytes
7c36125 f1dbb76 83f0319 abbaadf cd1a0ee 22350ec 7c36125 86cf0a3 e1bc29a c4e48d6 4b05098 22350ec 7c36125 8d38ee7 7c36125 7065361 8d38ee7 a1de7ad 2b40180 7c36125 1a9bf1d 8d38ee7 7c36125 8d38ee7 7c36125 a1d1c2a 7c36125 d71bb94 1a9bf1d 8d38ee7 7c36125 |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 |
import gradio as gr
import os
import sys
from pathlib import Path
models = [
"Yntec/photoMovieRealistic",
"Yntec/a-ZovyaRPGV3VAE",
"Yntec/a-ZoviaRPGArtistV2VAE",
"Yntec/InsaneRealisticCVAE",
"Yntec/Lunar",
"Yntec/LunarLuma",
"Yntec/QToriReloaded",
"Yntec/Ninja-Diffusers",
"Yntec/ChildrenStoriesAnime",
"Yntec/DucHaitenAnime768",
"Yntec/DucHaitenClassicAnime768",
"Yntec/WesternAnimation",
"Yntec/GOLDFish",
"Yntec/Rainbowsphere",
"Yntec/DreamAnything",
"Yntec/yabalMixTrue25D_v2_VAE",
"dreamlike-art/dreamlike-anime-1.0",
"Yntec/DucHaiten-Retro-Diffusers",
"Hemlok/QuinceMix",
"Yntec/KIDSILLUSTRATIONS",
"Yntec/COOLKIDSV2",
"Yntec/Pavo-Mix-Diffusers",
"Yntec/DucHaitenAIart-beta",
"Ojimi/anime-kawai-diffusion",
"LottePeisch/RevAnimated-Diffusers",
"Linaqruf/animagine-xl",
"Envvi/Inkpunk-Diffusion",
"Birchlabs/waifu-diffusion-xl-unofficial",
"digiplay/KawaiiRealisticAnimeMix_A0.3",
"digiplay/AI-infinity-V1-fp16",
]
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),
gr.Interface.load(f"models/{models[3]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[4]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[5]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[6]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[7]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[8]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[9]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[10]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[11]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[12]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[13]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[14]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[15]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[16]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[17]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[18]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[19]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[20]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[21]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[22]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[23]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[24]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[25]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[26]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[27]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[28]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[29]}",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.Tab("说明"):
gr.HTML(""" <title>多模型加速生成</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
<h1>输入文字生成图像</h1>
<br><h4>第一次导入模型可能需要200秒生成</h4>
<br><h4>之后一般只需20秒左右</h4>""")
with gr.Tab("内置模型解释"):
gr.HTML(""" <title>在内置模型中生成图片可能出现的问题</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
<h4>1. 使用Envvi/Inkpunk-Diffusion时，需要在提示词中加上“nvinkpunk”</h4>""")
with gr.Row():
with gr.Column(scale=100):
magic1=gr.Textbox(label="输入提示词：", lines=4, placeholder = "英文")
run=gr.Button("生成图像")
with gr.Row():
with gr.Column(scale=100):
model_name1 = gr.Dropdown(label="选择模型", 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):
gr.HTML("""<h1>没有想法？</h1>""")
input_text=gr.Textbox(label="输入提示词点子：",lines=2, placeholder = "英文")
use_short=gr.Button("使用精简提示词（使用框中提示词）")
see_prompts=gr.Button("扩充框内提示词")
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) |