Update app.py

app.py

@@ -76,6 +76,26 @@ from huggingface_hub import login
 
 is_colab = utils.is_google_colab()
 
+from share_btn import community_icon_html, loading_icon_html, share_js
+
+from huggingface_hub import login
+login()
+
+from huggingface_hub.inference_api import InferenceApi
+inference = InferenceApi(repo_id="bert-base-uncased", token=API_TOKEN)
+
+dataset = load_dataset("Guizmus/AnimeChanStyle")
+
+sys.path.append('src/blip')
+sys.path.append('src/clip')
+
+pipeline = DiffusionPipeline.from_pretrained("flax/waifu-diffusion")
+pipeline = DiffusionPipeline.from_pretrained("flax/Cyberpunk-Anime-Diffusion")
+pipeline = DiffusionPipeline.from_pretrained("technillogue/waifu-diffusion")
+pipeline = DiffusionPipeline.from_pretrained("svjack/Stable-Diffusion-Pokemon-en")
+pipeline = DiffusionPipeline.from_pretrained("AdamOswald1/Idk")
+pipeline = DiffusionPipeline.from_pretrained("katakana/2D-Mix")
+
 class Model:
     def __init__(self, name, path, prefix):
         self.name = name
@@ -90,151 +110,151 @@ models = [
     Model("Archer", "nitrosocke/archer-diffusion", "archer style"),
     Model("Elden Ring", "nitrosocke/elden-ring-diffusion", "elden ring style"),
     Model("Spider-Verse", "nitrosocke/spider-verse-diffusion", "spiderverse style"),
-     Model("Modern Disney", "nitrosocke/modern-disney-diffusion", "modern disney style"),
-     Model("Classic Disney", "nitrosocke/classic-anim-diffusion", "classic disney style"),
-     Model("Waifu", "hakurei/waifu-diffusion", ""),
-     Model("Pokémon", "lambdalabs/sd-pokemon-diffusers", "pokemon style"),
-     Model("Pokémon", "svjack/Stable-Diffusion-Pokemon-en", "pokemon style"),
-     Model("Pony Diffusion", "AstraliteHeart/pony-diffusion", "pony style"),
-     Model("Robo Diffusion", "nousr/robo-diffusion", "robo style"),
-     Model("Cyberpunk Anime", "DGSpitzer/Cyberpunk-Anime-Diffusion, flax/Cyberpunk-Anime-Diffusion", "cyberpunk style"),
-     Model("Cyberpunk Anime", "DGSpitzer/Cyberpunk-Anime-Diffusion", "cyberpunk style"),
-     Model("Cyberpunk Anime", "flax/Cyberpunk-Anime-Diffusion", "cyberpunk style"),
-     Model("Cyberware", "Eppinette/Cyberware", "cyberware"),
-     Model("Tron Legacy", "dallinmackay/Tron-Legacy-diffusion", "trnlgcy"),
-     Model("Waifu", "flax/waifu-diffusion", ""),
-     Model("Dark Souls", "Guizmus/DarkSoulsDiffusion", "dark souls style"),
-     Model("Waifu", "technillogue/waifu-diffusion", ""),
-     Model("Ouroborus", "Eppinette/Ouroboros", "m_ouroboros style"),
-     Model("Ouroborus alt", "Eppinette/Ouroboros", "m_ouroboros"),
-     Model("Waifu", "Eppinette/Mona", "Mona"),
-     Model("Waifu", "Eppinette/Mona", "Mona Woman"),
-     Model("Waifu", "Eppinette/Mona", "Mona Genshin"),
-     Model("Genshin", "Eppinette/Mona", "Mona"),
-     Model("Genshin", "Eppinette/Mona", "Mona Woman"),
-     Model("Genshin", "Eppinette/Mona", "Mona Genshin"),
-     Model("Space Machine", "rabidgremlin/sd-db-epic-space-machine", "EpicSpaceMachine"),
-     Model("Spacecraft", "rabidgremlin/sd-db-epic-space-machine", "EpicSpaceMachine"),
-     Model("TARDIS", "Guizmus/Tardisfusion", "Classic Tardis style"),
-     Model("TARDIS", "Guizmus/Tardisfusion", "Modern Tardis style"),
-     Model("TARDIS", "Guizmus/Tardisfusion", "Tardis Box style"),
-     Model("Spacecraft", "Guizmus/Tardisfusion", "Classic Tardis style"),
-     Model("Spacecraft", "Guizmus/Tardisfusion", "Modern Tardis style"),
-     Model("Spacecraft", "Guizmus/Tardisfusion", "Tardis Box style"),
-     Model("CLIP", "EleutherAI/clip-guided-diffusion", "CLIP"),
-     Model("Face Swap", "felixrosberg/face-swap", "faceswap"),
-     Model("Face Swap", "felixrosberg/face-swap", "faceswap with"),
-     Model("Face Swap", "felixrosberg/face-swap", "faceswapped"),
-     Model("Face Swap", "felixrosberg/face-swap", "faceswapped with"),
-     Model("Face Swap", "felixrosberg/face-swap", "face on"),
-     Model("Waifu", "Fampai/lumine_genshin_impact", "lumine_genshin"),
-     Model("Waifu", "Fampai/lumine_genshin_impact", "lumine"),
-     Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine Genshin"),
-     Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine_genshin"),
-     Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine_Genshin"),
-     Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine_genshin"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine_Genshin"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine Genshin"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "lumine"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Ganyu"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Woman"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Genshin"),
-     Model("Waifu", "sd-concepts-library/ganyu-genshin-impact", "Ganyu"),
-     Model("Waifu", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Woman"),
-     Model("Waifu", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Genshin"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "raiden_ei"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden Ei"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "Ei Genshin"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden Genshin"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden_Genshin"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "Ei_Genshin"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden"),
-     Model("Waifu", "Fampai/raiden_genshin_impact", "Ei"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden Ei"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "raiden_ei"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden Genshin"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Ei Genshin"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden_Genshin"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Ei_Genshin"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Ei"),
-     Model("Waifu", "Fampai/hutao_genshin_impact", "hutao_genshin"),
-     Model("Waifu", "Fampai/hutao_genshin_impact", "HuTao_Genshin"),
-     Model("Waifu", "Fampai/hutao_genshin_impact", "HuTao Genshin"),
-     Model("Waifu", "Fampai/hutao_genshin_impact", "HuTao"),
-     Model("Waifu", "Fampai/hutao_genshin_impact", "hutao_genshin"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "hutao_genshin"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "HuTao_Genshin"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "HuTao Genshin"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "HuTao"),
-     Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Female"),
-     Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "female"),
-     Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Woman"),
-     Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "woman"),
-     Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Girl"),
-     Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "girl"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "Female"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "female"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "Woman"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "woman"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "Girl"),
-     Model("Genshin", "Fampai/lumine_genshin_impact", "girl"),
-     Model("Genshin", "Eppinette/Mona", "Female"),
-     Model("Genshin", "Eppinette/Mona", "female"),
-     Model("Genshin", "Eppinette/Mona", "Woman"),
-     Model("Genshin", "Eppinette/Mona", "woman"),
-     Model("Genshin", "Eppinette/Mona", "Girl"),
-     Model("Genshin", "Eppinette/Mona", "girl"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Female"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "female"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Woman"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "woman"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Girl"),
-     Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "girl"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Female"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "female"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Woman"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "woman"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "Girl"),
-     Model("Genshin", "Fampai/raiden_genshin_impact", "girl"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "Female"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "female"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "Woman"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "woman"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "Girl"),
-     Model("Genshin", "Fampai/hutao_genshin_impact", "girl"),
-     Model("Waifu", "crumb/genshin-stable-inversion, yuiqena/GenshinImpact, Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Genshin"),
-     Model("Waifu", "crumb/genshin-stable-inversion, yuiqena/GenshinImpact, Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Genshin Impact"),
-     Model("Genshin", "crumb/genshin-stable-inversion, yuiqena/GenshinImpact, Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", ""),
-     Model("Waifu", "crumb/genshin-stable-inversion", "Genshin"),
-     Model("Waifu", "crumb/genshin-stable-inversion", "Genshin Impact"),
-     Model("Genshin", "crumb/genshin-stable-inversion", ""),
-     Model("Waifu", "yuiqena/GenshinImpact", "Genshin"),
-     Model("Waifu", "yuiqena/GenshinImpact", "Genshin Impact"),
-     Model("Genshin", "yuiqena/GenshinImpact", ""),
-     Model("Waifu", "hakurei/waifu-diffusion, flax/waifu-diffusion, technillogue/waifu-diffusion, Guizmus/AnimeChanStyle, katakana/2D-Mix", ""),
-     Model("Pokémon", "lambdalabs/sd-pokemon-diffusers, svjack/Stable-Diffusion-Pokemon-en", "pokemon style"),
-     Model("Pokémon", "lambdalabs/sd-pokemon-diffusers, svjack/Stable-Diffusion-Pokemon-en", ""),
-     Model("Test", "AdamoOswald1/Idk", ""),
-     Model("Anime", "Guizmus/AnimeChanStyle", "AnimeChan Style"),
-     Model("Genshin", "Guizmus/AnimeChanStyle", "AnimeChan Style"),
-     Model("Waifu", "Guizmus/AnimeChanStyle", "AnimeChan Style"),
-     Model("Waifu", "Guizmus/AnimeChanStyle", "Genshin"),
-     Model("Waifu", "Guizmus/AnimeChanStyle", "Genshin Impact"),
-     Model("Genshin", "Guizmus/AnimeChanStyle", ""),
-     Model("Anime", "Guizmus/AnimeChanStyle", ""),
-     Model("Waifu", "Guizmus/AnimeChanStyle", ""),
-     Model("Anime", "Guizmus/AnimeChanStyle, katakana/2D-Mix", ""),
-     Model("Anime", "katakana/2D-Mix", "2D-Mix"),
-     Model("Genshin", "katakana/2D-Mix", "2D-Mix"),
-     Model("Waifu", "katakana/2D-Mix", "2D-Mix"),
-     Model("Waifu", "katakana/2D-Mix", "Genshin"),
-     Model("Waifu", "katakana/2D-Mix", "Genshin Impact"),
-     Model("Genshin", "katakana/2D-Mix", ""),
-     Model("Anime", "katakana/2D-Mix", ""),
-     Model("Waifu", "katakana/2D-Mix", ""),
+    Model("Modern Disney", "nitrosocke/modern-disney-diffusion", "modern disney style"),
+    Model("Classic Disney", "nitrosocke/classic-anim-diffusion", "classic disney style"),
+    Model("Waifu", "hakurei/waifu-diffusion", ""),
+    Model("Pokémon", "lambdalabs/sd-pokemon-diffusers", "pokemon style"),
+    Model("Pokémon", "svjack/Stable-Diffusion-Pokemon-en", "pokemon style"),
+    Model("Pony Diffusion", "AstraliteHeart/pony-diffusion", "pony style"),
+    Model("Robo Diffusion", "nousr/robo-diffusion", "robo style"),
+    Model("Cyberpunk Anime", "DGSpitzer/Cyberpunk-Anime-Diffusion, flax/Cyberpunk-Anime-Diffusion", "cyberpunk style"),
+    Model("Cyberpunk Anime", "DGSpitzer/Cyberpunk-Anime-Diffusion", "cyberpunk style"),
+    Model("Cyberpunk Anime", "flax/Cyberpunk-Anime-Diffusion", "cyberpunk style"),
+    Model("Cyberware", "Eppinette/Cyberware", "cyberware"),
+    Model("Tron Legacy", "dallinmackay/Tron-Legacy-diffusion", "trnlgcy"),
+    Model("Waifu", "flax/waifu-diffusion", ""),
+    Model("Dark Souls", "Guizmus/DarkSoulsDiffusion", "dark souls style"),
+    Model("Waifu", "technillogue/waifu-diffusion", ""),
+    Model("Ouroborus", "Eppinette/Ouroboros", "m_ouroboros style"),
+    Model("Ouroborus alt", "Eppinette/Ouroboros", "m_ouroboros"),
+    Model("Waifu", "Eppinette/Mona", "Mona"),
+    Model("Waifu", "Eppinette/Mona", "Mona Woman"),
+    Model("Waifu", "Eppinette/Mona", "Mona Genshin"),
+    Model("Genshin", "Eppinette/Mona", "Mona"),
+    Model("Genshin", "Eppinette/Mona", "Mona Woman"),
+    Model("Genshin", "Eppinette/Mona", "Mona Genshin"),
+    Model("Space Machine", "rabidgremlin/sd-db-epic-space-machine", "EpicSpaceMachine"),
+    Model("Spacecraft", "rabidgremlin/sd-db-epic-space-machine", "EpicSpaceMachine"),
+    Model("TARDIS", "Guizmus/Tardisfusion", "Classic Tardis style"),
+    Model("TARDIS", "Guizmus/Tardisfusion", "Modern Tardis style"),
+    Model("TARDIS", "Guizmus/Tardisfusion", "Tardis Box style"),
+    Model("Spacecraft", "Guizmus/Tardisfusion", "Classic Tardis style"),
+    Model("Spacecraft", "Guizmus/Tardisfusion", "Modern Tardis style"),
+    Model("Spacecraft", "Guizmus/Tardisfusion", "Tardis Box style"),
+    Model("CLIP", "EleutherAI/clip-guided-diffusion", "CLIP"),
+    Model("Face Swap", "felixrosberg/face-swap", "faceswap"),
+    Model("Face Swap", "felixrosberg/face-swap", "faceswap with"),
+    Model("Face Swap", "felixrosberg/face-swap", "faceswapped"),
+    Model("Face Swap", "felixrosberg/face-swap", "faceswapped with"),
+    Model("Face Swap", "felixrosberg/face-swap", "face on"),
+    Model("Waifu", "Fampai/lumine_genshin_impact", "lumine_genshin"),
+    Model("Waifu", "Fampai/lumine_genshin_impact", "lumine"),
+    Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine Genshin"),
+    Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine_genshin"),
+    Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine_Genshin"),
+    Model("Waifu", "Fampai/lumine_genshin_impact", "Lumine"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine_genshin"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine_Genshin"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "Lumine Genshin"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "lumine"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Ganyu"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Woman"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Genshin"),
+    Model("Waifu", "sd-concepts-library/ganyu-genshin-impact", "Ganyu"),
+    Model("Waifu", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Woman"),
+    Model("Waifu", "sd-concepts-library/ganyu-genshin-impact", "Ganyu Genshin"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "raiden_ei"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden Ei"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "Ei Genshin"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden Genshin"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden_Genshin"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "Ei_Genshin"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "Raiden"),
+    Model("Waifu", "Fampai/raiden_genshin_impact", "Ei"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden Ei"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "raiden_ei"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden Genshin"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Ei Genshin"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Raiden_Genshin"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Ei_Genshin"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Ei"),
+    Model("Waifu", "Fampai/hutao_genshin_impact", "hutao_genshin"),
+    Model("Waifu", "Fampai/hutao_genshin_impact", "HuTao_Genshin"),
+    Model("Waifu", "Fampai/hutao_genshin_impact", "HuTao Genshin"),
+    Model("Waifu", "Fampai/hutao_genshin_impact", "HuTao"),
+    Model("Waifu", "Fampai/hutao_genshin_impact", "hutao_genshin"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "hutao_genshin"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "HuTao_Genshin"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "HuTao Genshin"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "HuTao"),
+    Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Female"),
+    Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "female"),
+    Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Woman"),
+    Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "woman"),
+    Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Girl"),
+    Model("Genshin", "Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "girl"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "Female"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "female"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "Woman"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "woman"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "Girl"),
+    Model("Genshin", "Fampai/lumine_genshin_impact", "girl"),
+    Model("Genshin", "Eppinette/Mona", "Female"),
+    Model("Genshin", "Eppinette/Mona", "female"),
+    Model("Genshin", "Eppinette/Mona", "Woman"),
+    Model("Genshin", "Eppinette/Mona", "woman"),
+    Model("Genshin", "Eppinette/Mona", "Girl"),
+    Model("Genshin", "Eppinette/Mona", "girl"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Female"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "female"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Woman"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "woman"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "Girl"),
+    Model("Genshin", "sd-concepts-library/ganyu-genshin-impact", "girl"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Female"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "female"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Woman"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "woman"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "Girl"),
+    Model("Genshin", "Fampai/raiden_genshin_impact", "girl"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "Female"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "female"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "Woman"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "woman"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "Girl"),
+    Model("Genshin", "Fampai/hutao_genshin_impact", "girl"),
+    Model("Waifu", "crumb/genshin-stable-inversion, yuiqena/GenshinImpact, Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Genshin"),
+    Model("Waifu", "crumb/genshin-stable-inversion, yuiqena/GenshinImpact, Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", "Genshin Impact"),
+    Model("Genshin", "crumb/genshin-stable-inversion, yuiqena/GenshinImpact, Fampai/lumine_genshin_impact, Eppinette/Mona, sd-concepts-library/ganyu-genshin-impact, Fampai/raiden_genshin_impact, Fampai/hutao_genshin_impact", ""),
+    Model("Waifu", "crumb/genshin-stable-inversion", "Genshin"),
+    Model("Waifu", "crumb/genshin-stable-inversion", "Genshin Impact"),
+    Model("Genshin", "crumb/genshin-stable-inversion", ""),
+    Model("Waifu", "yuiqena/GenshinImpact", "Genshin"),
+    Model("Waifu", "yuiqena/GenshinImpact", "Genshin Impact"),
+    Model("Genshin", "yuiqena/GenshinImpact", ""),
+    Model("Waifu", "hakurei/waifu-diffusion, flax/waifu-diffusion, technillogue/waifu-diffusion, Guizmus/AnimeChanStyle, katakana/2D-Mix", ""),
+    Model("Pokémon", "lambdalabs/sd-pokemon-diffusers, svjack/Stable-Diffusion-Pokemon-en", "pokemon style"),
+    Model("Pokémon", "lambdalabs/sd-pokemon-diffusers, svjack/Stable-Diffusion-Pokemon-en", ""),
+    Model("Test", "AdamoOswald1/Idk", ""),
+    Model("Anime", "Guizmus/AnimeChanStyle", "AnimeChan Style"),
+    Model("Genshin", "Guizmus/AnimeChanStyle", "AnimeChan Style"),
+    Model("Waifu", "Guizmus/AnimeChanStyle", "AnimeChan Style"),
+    Model("Waifu", "Guizmus/AnimeChanStyle", "Genshin"),
+    Model("Waifu", "Guizmus/AnimeChanStyle", "Genshin Impact"),
+    Model("Genshin", "Guizmus/AnimeChanStyle", ""),
+    Model("Anime", "Guizmus/AnimeChanStyle", ""),
+    Model("Waifu", "Guizmus/AnimeChanStyle", ""),
+    Model("Anime", "Guizmus/AnimeChanStyle, katakana/2D-Mix", ""),
+    Model("Anime", "katakana/2D-Mix", "2D-Mix"),
+    Model("Genshin", "katakana/2D-Mix", "2D-Mix"),
+    Model("Waifu", "katakana/2D-Mix", "2D-Mix"),
+    Model("Waifu", "katakana/2D-Mix", "Genshin"),
+    Model("Waifu", "katakana/2D-Mix", "Genshin Impact"),
+    Model("Genshin", "katakana/2D-Mix", ""),
+    Model("Anime", "katakana/2D-Mix", ""),
+    Model("Waifu", "katakana/2D-Mix", ""),
     Model("Beeple", "riccardogiorato/beeple-diffusion", "beeple style "),
     Model("Avatar", "riccardogiorato/avatar-diffusion", "avatartwow style "),
     Model("Poolsuite", "prompthero/poolsuite", "poolsuite style ")
@@ -266,6 +286,17 @@ current_model_path = current_model.path
 
 if is_colab:
   pipe = StableDiffusionPipeline.from_pretrained(current_model.path, torch_dtype=torch.float16, scheduler=scheduler)
+  pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", torch_type=torch.float16, revision="fp16")
+  pipe = StableDiffusionPipeline.from_pretrained(current_model, torch_dtype=torchfloat, revision="fp16")
+  gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
+  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True, revision="fp16", torch_dtype=torch.float16).to("cuda")
+  pipe = StableDiffusionPipeline.from_pretrained(current_model.path, torch_dtype=torch.float16)
+  pipeline = DiffusionPipeline.from_pretrained("flax/waifu-diffusion")
+  pipeline = DiffusionPipeline.from_pretrained("flax/Cyberpunk-Anime-Diffusion")
+  pipeline = DiffusionPipeline.from_pretrained("technillogue/waifu-diffusion")
+  pipeline = DiffusionPipeline.from_pretrained("svjack/Stable-Diffusion-Pokemon-en")
+  pipeline = DiffusionPipeline.from_pretrained("AdamOswald1/Idk")
+  pipeline = DiffusionPipeline.from_pretrained("katakana/2D-Mix")
 
 else: # download all models
   vae = AutoencoderKL.from_pretrained(current_model.path, subfolder="vae", torch_dtype=torch.float16)
@@ -331,8 +362,7 @@ def txt_to_img(model_path, prompt, neg_prompt, guidance, steps, width, height, g
       width = width,
       height = height,
       generator = generator)
-    
-    return replace_nsfw_images(result)
+ 
 
 def img_to_img(model_path, prompt, neg_prompt, img, strength, guidance, steps, width, height, generator=None):
 
@@ -367,14 +397,6 @@ def img_to_img(model_path, prompt, neg_prompt, img, strength, guidance, steps, w
         height = height,
         generator = generator)
         
-    return replace_nsfw_images(result)
-
-def replace_nsfw_images(results):
-    for i in range(len(results.images)):
-      if results.nsfw_content_detected[i]:
-        results.images[i] = Image.open("nsfw.png")
-    return results.images[0]
-
 css = """.finetuned-diffusion-div div{display:inline-flex;align-items:center;gap:.8rem;font-size:1.75rem}.finetuned-diffusion-div div h1{font-weight:900;margin-bottom:7px}.finetuned-diffusion-div p{margin-bottom:10px;font-size:94%}a{text-decoration:underline}.tabs{margin-top:0;margin-bottom:0}#gallery{min-height:20rem}
 """
 with gr.Blocks(css=css) as demo:
@@ -445,15 +467,6 @@ with gr.Blocks(css=css) as demo:
     prompt.submit(inference, inputs=inputs, outputs=image_out)
     generate.click(inference, inputs=inputs, outputs=image_out)
 
-    ex = gr.Examples([
-        [models[1].name, "Neon techno-magic robot with spear pierces an ancient beast, hyperrealism, no blur, 4k resolution, ultra detailed", 7.5, 50],
-        [models[1].name, "halfturn portrait of a big crystal face of a beautiful abstract ancient Egyptian elderly shaman woman, made of iridescent golden crystals, half - turn, bottom view, ominous, intricate, studio, art by anthony macbain and greg rutkowski and alphonse mucha, concept art, 4k, sharp focus", 7.5, 25],
-    ], [model_name, prompt, guidance, steps, seed], image_out, inference, cache_examples=False)
-
-    gr.HTML("""
-       <p>Models by <a href="https://huggingface.co/riccardogiorato">@riccardogiorato</a><br></p>
-    """)
-
 if not is_colab:
   demo.queue(concurrency_count=1)
 demo.launch(debug=is_colab, share=is_colab)