Spaces:
Runtime error
Runtime error
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", | |
###################################################### | |
#"digiplay/PersonaStyleCheckpoint", | |
#"darkstorm2150/Protogen_Infinity_Official_Release", | |
#"PeggyWang/openjourney-v2", | |
#"darkstorm2150/Protogen_x3.4_Official_Release", | |
#"stablediffusionapi/deliberateappfactory", #236 | |
#"digiplay/CrossoverMix_v2", | |
#"stablediffusionapi/spybg", | |
#"stablediffusionapi/dreamshaper-v6", #239 | |
#"stablediffusionapi/the-ally", | |
#"darkstorm2150/Protogen_x5.8_Official_Release", | |
#coreco/seek.art_MEGA", | |
#"digiplay/BlankCanvas_v1", #07.11 | |
#"digiplay/OnlyAnime_v2.3", | |
#"Korakoe/OpenNiji", | |
#"digiplay/Photon_v1", | |
#"digiplay/Pika_v2", | |
#"digiplay/RealCartoon3D_F16full_v3.1", #254 | |
#"digiplay/realidefmix_3.5VAE", | |
#"digiplay/realmixUnrealjourney_v1", | |
#"digiplay/SyncMix_v1.5", | |
#"digiplay/TWingshadow_v1.2", | |
#"digiplay/V3_by_Hans_Asian", | |
#"digiplay/whatamix_v1", | |
# | |
#"digiplay/2K", #216 | |
#"digiplay/AIGEN_v1.4_diffusers", | |
#"digiplay/BrickAndMortarMix_v2.0_diffusers", #224 | |
#"digiplay/BeautyFool_v1.2VAE_pruned", | |
#"digiplay/breakdomainrealistic_R2333", | |
# "digiplay/CCTV2.5d_v1", #219 | |
# "digiplay/ChikMix_V3", #253 | |
# "stablediffusionapi/chilledremixsazyou-r", #195 | |
# "digiplay/CityEdge_StyleMix_v1.44", | |
# "stablediffusionapi/dalcefopainting2", #199 | |
# "digiplay/EdisonNilMix_v1", #07.10 | |
# "digiplay/DiamondCoalMix_v2_pruned_diffusers", | |
# "digiplay/DreamShaper_7", #259 | |
# "digiplay/elegantEntropy_v1.1", #221 | |
# "digiplay/EtherRealMix_LUX2", | |
# "digiplay/KawaiiRealisticAnimeMix_A0.3", | |
# "digiplay/highQualityCGMIX_v1", | |
# "digiplay/HIMAWARI_v1", | |
# "digiplay/Hodgepodge_v2.1", #217 | |
# "digiplay/illustro1stEdition_illustroV1", #214 | |
# "digiplay/Juggernaut_final", #07.11 | |
# "digiplay/Landscape_PhotoReal_v1", | |
# "digiplay/LuckyStrikeMix0.2Realistic", #07.10 | |
# "digiplay/Matrix_Stellar_VAE_v1", | |
# "digiplay/PrefixRealisticMix_v1", | |
# "digiplay/RealEpicMajicRevolution_v1", #07.11 | |
# "digiplay/ShampooMix_4", #252 | |
# "digiplay/SoapMix2.5D_v1", | |
# "digiplay/ZemiHR_v2_diffusers", | |
# | |
#"Redamancy2299/dreambooth", | |
#"Lykon/DreamShaper", #240 | |
#"trysem/DreamShaper-3.3", | |
#"HusseinHE/hussein-deliberate-1000steps", #237 | |
#"stablediffusionapi/majicmixfantasy", | |
#"stablediffusionapi/majicmixsombre", #247 | |
#"wavymulder/modelshoot", | |
#"digiplay/ChillyMix_v1", #215 | |
#"stablediffusionapi/foto-assisted-diffusion", #197 | |
#"wavymulder/portraitplus", | |
#"stablediffusionapi/chilloutmix-4264", | |
#"stablediffusionapi/product-design", #194 | |
# "kandinsky-community/kandinsky-2-1", #251 | |
# | |
#"digiplay/2.5DSET_diffusers", #227 | |
#"digiplay/2-KWI", #213 | |
#"digiplay/alstroemeriaMix_v1", | |
#"wavymulder/Analog-Diffusion", | |
#"digiplay/AniRealityMix_v1", #257 | |
#"digiplay/ARRealVX1.1", | |
#"digiplay/BadAnime_v1", | |
#"digiplay/BasilKorea_v2", #07.11 | |
#"digiplay/bluePencilRealistic_v01", | |
#"digiplay/bra_v40_diffusers", | |
#"digiplay/Burger_Mix_semiR2Lite", #222 | |
#"digiplay/calicomixreal_v2.0_diffusers", | |
#"digiplay/CampurSari_Gen1", | |
#"digiplay/cocotifacute_v1", #07.10 | |
#"digiplay/cosfMix_v1", #223 | |
#"digiplay/CounterMix_v2", #211 | |
#"digiplay/CuriousMerge2.5D_v5", | |
#"digiplay/dosmix", | |
#"stablediffusionapi/droodlyrielv15", | |
#"digiplay/fantexi_v0.7", | |
#"digiplay/fishmix_other_v1", | |
#"digiplay/FormCleansingMix_v1", #228 | |
#"digiplay/FumizukiMix_v1", | |
#"digiplay/helloworld_v3", | |
#"digiplay/HenmixArt_v1", | |
#"digiplay/ISOmix_v3.22", | |
#"digiplay/kencanmix_v2.0beta", | |
#"wavymulder/lomo-diffusion", | |
#"stablediffusionapi/majicmixv5", #192 | |
#"digiplay/mecha_musume_vivid_soft", | |
#"digiplay/MiracleMixGlitter_v1", | |
#"digiplay/MixTape_RocknRoll_v3punk_bake_fp16", | |
#"digiplay/NextPhoto_v1", | |
#"digiplay/Noosphere_v3", | |
#"digiplay/nk15_diffusers", #230 | |
#"digiplay/PeachMixsRelistic_R0", #262 | |
# "wavymulder/timeless-diffusion", | |
# "digiplay/WhiteDreamyHillMix_v1", #220 | |
# "digiplay/ya3p_VAE", #258 | |
# | |
#"DucHaiten/DucHaitenAnime", | |
#"DucHaiten/DucHaitenAIart", | |
#"Manseo/Colorful-v4.5-Plus", #244 | |
#"Guizmus/SDArt_ChaosAndOrder", | |
#"DucHaiten/DH_ClassicAnime", | |
#"stablediffusionapi/disneypixar", | |
#"johnslegers/epic-diffusion-v1.1", | |
#"emilianJR/epiCRealism", | |
#"johnslegers/epic-diffusion", | |
#"digiplay/endlessMixRenatus_v1.1", #07.10 | |
#"digiplay/fantasticAnime_diffusers", | |
#"stablediffusionapi/ghostmix", | |
#"Duskfallcrew/EpicMix_Realism", | |
#"nitrosocke/Nitro-Diffusion", | |
#"prompthero/openjourney", | |
#"Guizmus/SDArt_something", | |
# "DucHaiten/DucHaiten-StyleLikeMe", | |
# "ddPn08/subtly", #250 | |
# "22h/vintedois-diffusion-v0-1", | |
# | |
#"circulus/sd-anireal-v2.7", | |
#"0xJustin/Dungeons-and-Diffusion", | |
# "Guizmus/SDArt_AliceInDiffusionLand", | |
# "stablediffusionapi/realistic-vision-v20-2047", | |
# "redstonehero/RPG-v5-itr17_A10T", | |
# | |
# "stablediffusionapi/camelliamix25d", | |
# "Guizmus/SDArt_cosmichorrors", | |
# "DGSpitzer/DGSpitzer-Art-Diffusion", | |
# "stablediffusionapi/emotion-puppeteer-v2", | |
#"stablediffusionapi/fengjing", | |
# "stablediffusionapi/fuwafuwamix", | |
# "Fred99774/girlnew1", | |
# "stablediffusionapi/majicmixrealistic", | |
# "badmonk/nxka", | |
#"ItsJayQz/SynthwavePunk-v2", | |
#"zhyemmmm/ToonYou", | |
#"stablediffusionapi/uber-realistic-merge", | |
#"stablediffusionapi/vne732h9dh4", | |
#"stablediffusionapi/wand-magic2", | |
#"stablediffusionapi/waifu-journey-2", | |
#"stablediffusionapi/zovya", | |
#"Guizmus/SDArt_cosmichorrors768", | |
#"stablediffusionapi/counterfeit-v30", | |
#"stablediffusionapi/amireal", | |
##"JamesFlare/pastel-mix", #"andite/pastel-mix", | |
#"stablediffusionapi/rev-anim", | |
#"aipicasso/picasso-diffusion-1-1", | |
#"xiaolxl/Gf_style2", | |
#"circulus/sd-semireal-v2.8", | |
#"Crosstyan/BPModel", #07.11 | |
#"digiplay/Dusk-1", | |
#"ogkalu/Comic-Diffusion", | |
#"Guizmus/SDArt_ChaosAndOrder768", | |
#"gsdf/Counterfeit-V2.0", | |
#"dwancin/memoji", #07.11 | |
#"nousr/robo-diffusion-2-base", | |
###"hakurei/waifu-diffusion", | |
#"WarriorMama777/AbyssOrangeMix2", | |
#"stablediffusionapi/abyssorangemix2nsfw", #200 | |
#"cag/anything-v3-1", | |
#"iZELX1/Anything-V3-X", | |
#"xyn-ai/anything-v4.0", #"andite/anything-v4.0", | |
#"D1b4l4p/AsianMix", | |
#"Fred99774/chilloutvlara", | |
#"aipicasso/cool-japan-diffusion-2-1-2", | |
#"stablediffusionapi/corneos-7th-heaven-m", #196 | |
#"DGSpitzer/Cyberpunk-Anime-Diffusion", | |
#"stablediffusionapi/dark-sushi-mix", | |
#"joachimsallstrom/Double-Exposure-Diffusion", | |
#"eimiss/EimisAnimeDiffusion_1.0v", | |
#"prompthero/funko-diffusion", | |
#"nitrosocke/Ghibli-Diffusion", | |
###"iZELX1/Grapefruit", | |
#"xiaolxl/GuoFeng3", | |
#"stablediffusionapi/tmnd-mix", | |
#"coder119/Vectorartz_Diffusion", #203 | |
#"WarriorMama777/AbyssOrangeMix", | |
#"AIARTCHAN/7pa", | |
#"JosephusCheung/ACertainModel", | |
#"JosephusCheung/ACertainThing", | |
#"AIARTCHAN/AbyssHellHero", | |
#"JosephusCheung/ACertainty", | |
#"AIARTCHAN/AbyssHellVer3", | |
#"AIARTCHAN/AbyssMapleVer3", | |
#"stablediffusionapi/abyssorangemixsfw", | |
#"AIARTCHAN/anidosmixV2", | |
#"stablediffusionapi/anime-model-v2", | |
#"kubanemil/AnyLORA", | |
#"stablediffusionapi/hc-anything-v3-vae", #231 | |
#"mm00/anything-v3.0-light", | |
#"stablediffusionapi/anythingelse-v4", | |
#"stablediffusionapi/anything-v45-fixed", | |
#"stablediffusionapi/anything-v5", | |
#"nitrosocke/Arcane-Diffusion", | |
#"nitrosocke/archer-diffusion", | |
#"stablediffusionapi/architecture-tuned-model", | |
#"WarriorMama777/BloodOrangeMix", | |
#"wavymulder/collage-diffusion", | |
#"stablediffusionapi/camelliamixline", | |
#"digiplay/chrysanthemumMix_v1", | |
#"digiplay/CiderMix_ciderR", #260 | |
#"Johnhex/Clam", #243 | |
#"stablediffusionapi/cosmic-babes", | |
#"digiplay/CoffeeDonut_v1", | |
#"stablediffusionapi/dark-sushi-25d", | |
#"digiplay/Defacta_v1_diffusers", #226 | |
### "WarriorMama777/EerieOrangeMix", | |
#"digiplay/DuelAnimeMix_v1", #225 | |
#"Envvi/Inkpunk-Diffusion", | |
#"digiplay/kotosmix_diffusers", #229 | |
#"stablediffusionapi/meinaalter", | |
#"Nacholmo/meinamixv7-diffusers", | |
#"stablediffusionapi/meinapastel", | |
#"AIARTCHAN/MIX-Pro-V4", | |
#"Lykon/NeverEnding-Dream", | |
#"stablediffusionapi/shirataki-mix", #191 | |
#"NoCrypt/SomethingV2_2", | |
#"NoCrypt/SomethingV2", | |
#"badmonk/sxzumi", | |
### "stablediffusionapi/three-delicacy", | |
### "stablediffusionapi/three-delicacy-wonto", | |
#"etherealxx/systemy-csrmodel-cutesexyrobutts", #"andite/cutesexyrobutts-diffusion", | |
#"sd-dreambooth-library/true-guweiz-style", # "andite/guweiz-diffusion", | |
#"stablediffusionapi/vector-art", #198 | |
#"digiplay/xxMix_4", | |
###"mio/hiten", #"andite/hiten-diffusion", | |
### "andite/mashuu-diffusion", | |
### "andite/mignon-diffusion", | |
#### "andite/mikapikazo-diffusion", | |
#### "andite/piromizu-diffusion", | |
#"digiplay/Zevinemix_v1.0/", | |
#"digiplay/AnaMix_v2", #07.11 | |
#"stablediffusionapi/animetestmodelv3", | |
#"yulet1de/anything", #232 | |
#"hakurei/artstation-diffusion", #07.11 | |
#"Fictiverse/Stable_Diffusion_BalloonArt_Model", | |
#"stablediffusionapi/bg-dream-irl", | |
#"stablediffusionapi/bg-dream-model-b", #193 | |
#"Rardilit/Ciffusion_v0.1", | |
#"circulus/sd-anireal-2d-v2", | |
#"circulus/sd-photoreal-v2.7", | |
#"circulus/sd-photoreal-photo-v2", | |
#"circulus/sd-anireal-2.5d-v2", | |
#"circulus/sd-anireal-v2.5", | |
#"circulus/sd-photoreal-semi-v2", | |
#"circulus/sd-photoreal-real-v2", | |
#"circulus/sd-photoreal-v2.5", | |
#"circulus/sd-anireal-3d-v2", | |
#"circulus/sd-anireal-v2.8", | |
#"nitrosocke/classic-anim-diffusion", | |
#"Conflictx/Complex-Lineart", #245 | |
#"sayakpaul/da-vinci-sd-pokemon", | |
#"nitrosocke/elden-ring-diffusion", | |
#"digiplay/EtherBluMix_1", #07.11 | |
#"digiplay/fantasticmix_v40_test", #261 | |
#"theintuitiveye/FantasyMix", | |
#"Fictiverse/Stable_Diffusion_FluidArt_Model", | |
#"nitrosocke/Future-Diffusion", | |
#"ItsJayQz/GTA5_Artwork_Diffusion", #205 | |
#"digiplay/hellopure_v2.23", | |
#"TheLastBen/hrrzg-style-768px", #246 | |
#"nevernotsean/IllustratedPaperMini", #242 | |
#"dallinmackay/JWST-Deep-Space-diffusion", | |
#"prompthero/linkedin-diffusion", | |
#"mann-e/mann-e_4_rev-0-1", #210 | |
#"ItsJayQz/Marvel_WhatIf_Diffusion", #206 | |
#"yuanbit/max-15-1e-6-1500", | |
#"MyneFactory/MF-Base", #248 | |
#"Fictiverse/Stable_Diffusion_Microscopic_model", #249 | |
#"nitrosocke/mo-di-diffusion", | |
#"luongphamit/NeverEnding-Dream2", #241 | |
#"lambdalabs/sd-naruto-diffusers", #201 | |
#"Vernon-2/output_test", | |
#"Fictiverse/Stable_Diffusion_PaperCut_Model", | |
##"bsuutari/path_to_saved_model", | |
#"bsuutari/path_to_saved_model_rafa", | |
#"digiplay/PlanetBumix_v1", | |
#"lambdalabs/sd-pokemon-diffusers", #202 | |
#"prompthero/poolsuite-diffusion", | |
#"digiplay/RealismEngine_v1", | |
#"nitrosocke/redshift-diffusion", | |
#"nitrosocke/redshift-diffusion-768", | |
#"nousr/robo-diffusion", | |
#"digiplay/SDVN1-Real_v1", #255 | |
#"nitrosocke/spider-verse-diffusion", | |
##"runwayml/stable-diffusion-v1-5", | |
#"nicky007/stable-diffusion-logo-fine-tuned", | |
#"stablediffusionapi/three-delicacy", #233 | |
#"stablediffusionapi/three-delicacy-wonto", #234 | |
#"naclbit/trinart_stable_diffusion_v2", | |
#"dallinmackay/Tron-Legacy-diffusion", | |
#"digiplay/unstableDiffusersYamerMIX_v3", | |
#dallinmackay/Van-Gogh-diffusion", | |
#"Fictiverse/Stable_Diffusion_VoxelArt_Model", #204 | |
#"wavymulder/wavyfusion", | |
#"CompVis/stable-diffusion-v1-3", #207 | |
#"CompVis/stable-diffusion-v1-2", #208 | |
#"CompVis/stable-diffusion-v1-1", #209 | |
] | |
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), | |
] | |
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.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=400px"): | |
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) |