blitz_diffusion / app.py
Yntec's picture
Upload 4 files
6adca2a
raw
history blame
20.1 kB
import gradio as gr
import os
import sys
from pathlib import Path
models = [
"Nacholmo/meinamixv7-diffusers",
"DucHaiten/DH_ClassicAnime",
"DucHaiten/DucHaitenAnime",
"DucHaiten/DucHaitenAIart",
"DucHaiten/DucHaitenDreamWorld",
"DucHaiten/DucHaitenJourney",
"DucHaiten/DucHaiten-StyleLikeMe",
"DucHaiten/DucHaitenSuperCute",
"Hius/DreamFul-V2",
"theintuitiveye/HARDblend",
"hassanblend/hassanblend1.4",
"Hemlok/QuinceMix",
"coreco/seek.art_MEGA",
"stablediffusionapi/all-526-animated",
"stablediffusionapi/all-526",
"stablediffusionapi/cyberrealistic",
"SG161222/Realistic_Vision_V1.4",
"claudfuen/photorealistic-fuen-v1",
"robotjung/SemiRealMix",
"prompthero/funko-diffusion",
"prompthero/midjourney-v4-diffusion",
"prompthero/openjourney-v4",
"FredZhang7/paint-journey-v2",
"x67/shortjourney",
"Joeythemonster/anything-midjourney-v-4-1",
"circulus/sd-semireal-v2.8",
"circulus/sd-anireal-v2.7",
"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",
"stablediffusionapi/spybg",
"darkstorm2150/Protogen_v2.2_Official_Release",
"darkstorm2150/Protogen_Infinity_Official_Release",
"darkstorm2150/Protogen_x3.4_Official_Release",
"darkstorm2150/Protogen_x5.8_Official_Release",
"wavymulder/Analog-Diffusion",
"wavymulder/collage-diffusion",
"wavymulder/lomo-diffusion",
"wavymulder/modelshoot",
"wavymulder/portraitplus",
"wavymulder/timeless-diffusion",
"ItsJayQz/SynthwavePunk-v2",
"gsdf/Counterfeit-V2.0",
"stablediffusionapi/counterfeit-v30",
"dreamlike-art/dreamlike-anime-1.0",
"dreamlike-art/dreamlike-diffusion-1.0",
"dreamlike-art/dreamlike-photoreal-2.0",
"stablediffusionapi/amireal",
"Korakoe/OpenNiji",
#"JamesFlare/pastel-mix", #"andite/pastel-mix",
"aipicasso/picasso-diffusion-1-1",
"stablediffusionapi/rev-anim",
"LottePeisch/RevAnimated-Diffusers",
"nousr/robo-diffusion-2-base",
"stablediffusionapi/the-ally",
"stablediffusionapi/tmnd-mix",
##"hakurei/waifu-diffusion",
"WarriorMama777/AbyssOrangeMix2",
"cag/anything-v3-1",
"iZELX1/Anything-V3-X",
"xyn-ai/anything-v4.0", #"andite/anything-v4.0",
"stablediffusionapi/chillout-app-factory",
"nitrosocke/classic-anim-diffusion",
"aipicasso/cool-japan-diffusion-2-1-2",
"DGSpitzer/Cyberpunk-Anime-Diffusion",
"stablediffusionapi/dark-sushi-mix",
"joachimsallstrom/Double-Exposure-Diffusion",
"eimiss/EimisAnimeDiffusion_1.0v",
"nitrosocke/Ghibli-Diffusion",
"xiaolxl/Gf_style2",
###"iZELX1/Grapefruit",
"xiaolxl/GuoFeng3",
"stablediffusionapi/meinapastel",
"WarriorMama777/AbyssOrangeMix",
"AIARTCHAN/7pa",
"JosephusCheung/ACertainModel",
"JosephusCheung/ACertainThing",
"JosephusCheung/ACertainty",
"AIARTCHAN/AbyssHellHero",
"AIARTCHAN/AbyssMapleVer3",
"AIARTCHAN/AbyssHellVer3",
"AIARTCHAN/anidosmixV2",
"Ojimi/anime-kawai-diffusion",
"kubanemil/AnyLORA",
"mm00/anything-v3.0-light",
"stablediffusionapi/anythingelse-v4",
"stablediffusionapi/anything-v5",
"nitrosocke/Arcane-Diffusion",
"nitrosocke/archer-diffusion",
"D1b4l4p/AsianMix",
"WarriorMama777/BloodOrangeMix",
"stablediffusionapi/camelliamix25d",
"stablediffusionapi/camelliamixline",
"stablediffusionapi/cosmic-babes",
"stablediffusionapi/dark-sushi-25d",
## "WarriorMama777/EerieOrangeMix",
"stablediffusionapi/icomix-2",
"Envvi/Inkpunk-Diffusion",
"dallinmackay/JWST-Deep-Space-diffusion",
"stablediffusionapi/meinaalter",
"AIARTCHAN/MIX-Pro-V4",
"Lykon/NeverEnding-Dream",
"NoCrypt/SomethingV2_2",
"NoCrypt/SomethingV2",
## "stablediffusionapi/three-delicacy",
## "stablediffusionapi/three-delicacy-wonto",
"etherealxx/systemy-csrmodel-cutesexyrobutts", #"andite/cutesexyrobutts-diffusion",
"sd-dreambooth-library/true-guweiz-style", # "andite/guweiz-diffusion",
###"mio/hiten", #"andite/hiten-diffusion",
### "andite/mashuu-diffusion",
### "andite/mignon-diffusion",
### "andite/mikapikazo-diffusion",
### "andite/piromizu-diffusion",
"Fictiverse/Stable_Diffusion_BalloonArt_Model",
"Rardilit/Ciffusion_v0.1",
"nitrosocke/elden-ring-diffusion",
"Fictiverse/Stable_Diffusion_FluidArt_Model",
"nitrosocke/Future-Diffusion",
"prompthero/linkedin-diffusion",
"nitrosocke/mo-di-diffusion",
"Fictiverse/Stable_Diffusion_PaperCut_Model",
"AstraliteHeart/pony-diffusion",
"prompthero/poolsuite-diffusion",
"SG161222/Realistic_Vision_V1.4",
"nitrosocke/redshift-diffusion",
"nitrosocke/redshift-diffusion-768",
"nousr/robo-diffusion",
"nitrosocke/spider-verse-diffusion",
#"runwayml/stable-diffusion-v1-5",
"naclbit/trinart_stable_diffusion_v2",
"dallinmackay/Tron-Legacy-diffusion",
"dallinmackay/Van-Gogh-diffusion",
"wavymulder/wavyfusion"
]
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),
gr.Interface.load(f"models/{models[30]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[31]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[32]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[33]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[34]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[35]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[36]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[37]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[38]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[39]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[40]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[41]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[42]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[43]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[44]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[45]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[46]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[47]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[48]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[49]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[50]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[51]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[52]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[53]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[54]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[55]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[56]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[57]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[58]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[59]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[60]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[61]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[62]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[63]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[64]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[65]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[66]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[67]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[68]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[69]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[70]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[71]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[72]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[73]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[74]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[75]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[76]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[77]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[78]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[79]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[80]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[81]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[82]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[83]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[84]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[85]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[86]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[87]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[88]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[89]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[90]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[91]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[92]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[93]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[94]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[95]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[96]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[97]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[98]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[99]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[100]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[101]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[102]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[103]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[104]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[105]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[106]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[107]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[108]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[109]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[110]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[111]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[112]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[113]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[114]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[115]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[116]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[117]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[118]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[119]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[120]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[121]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[122]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[123]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[124]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[125]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[126]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[127]}",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}")),gr.update(label=(f"{current_model}")),gr.update(label=(f"{current_model}")),gr.update(label=(f"{current_model}")),gr.update(label=(f"{current_model}")),gr.update(label=(f"{current_model}")),gr.update(label=(f"{current_model}")),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("""<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8" />
<meta name="twitter:card" content="player"/>
<meta name="twitter:site" content=""/>
<meta name="twitter:player" content="https://omnibus-maximum-multiplier-places.hf.space"/>
<meta name="twitter:player:stream" content="https://omnibus-maximum-multiplier-places.hf.space"/>
<meta name="twitter:player:width" content="100%"/>
<meta name="twitter:player:height" content="600"/>
<meta property="og:title" content="Embedded Live Viewer"/>
<meta property="og:description" content="Tweet Genie - A Huggingface Space"/>
<meta property="og:image" content="https://cdn.glitch.global/80dbe92e-ce75-44af-84d5-74a2e21e9e55/omnicard.png?v=1676772531627"/>
<!--<meta http-equiv="refresh" content="0; url=https://huggingface.co/spaces/corbt/tweet-genie">-->
</head>
</html>
""")
with gr.Row():
with gr.Tab("Title"):
gr.HTML(""" <title>Minimum Multiplier</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
<h1>Fill the Textbox at the bottom and click Generate Image</h1>
<br><h4>The first time you load a model it takes 160 seconds</h4>
<br><h4>But after it loads each image takes 10 seconds to generate!</h4>
""")
with gr.Tab("Description"):
gr.HTML("""<div style="text-align:center;">
<h4>As many Text-to-Image Models as I can fit here</h4><br>
<h4>Suggest more up in the "Community" button</h4>
</div>""")
with gr.Tab("Tools"):
with gr.Tab("View"):
with gr.Row():
with gr.Column(style="width=50%, height=70%"):
gr.Pil(label="Crop")
with gr.Column(style="width=50%, height=70%"):
gr.Pil(label="Crop")
with gr.Tab("Draw"):
with gr.Column(style="width=50%, height=70%"):
gr.Pil(label="Crop")
with gr.Column(style="width=50%, height=70%"):
gr.Pil(label="Draw")
gr.ImagePaint(label="Draw")
with gr.Tab("Text"):
with gr.Row():
with gr.Column(scale=50):
gr.Textbox(label="", lines=8, interactive=True)
with gr.Column(scale=50):
gr.Textbox(label="", lines=8, interactive=True)
with gr.Tab("Color Picker"):
with gr.Row():
with gr.Column(scale=50):
gr.ColorPicker(label="Color", interactive=True)
with gr.Column(scale=50):
gr.ImagePaint(label="Draw", interactive=True)
with gr.Row():
with gr.Column():
input_text=gr.Textbox(label="Prompt Idea",lines=2)
#Model selection dropdown
model_name1 = gr.Dropdown(show_label=False, choices=[m for m in models], type="index", value=current_model, interactive=True)
with gr.Column():
use_short=gr.Button("Use Short Prompt")
see_prompts=gr.Button("Extend Idea")
run=gr.Button("Generate Image")
with gr.Tab("Main"):
with gr.Row():
output1=gr.Image(label=(f"{current_model}"))
with gr.Row():
magic1=gr.Textbox(lines=4)
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)