Update app.py

app.py

@@ -2,7 +2,7 @@ from diffusers import DiffusionPipeline
 import torch
 import gradio as gr
 from PIL import Image
-import os, random, gc, re, json, time, shutil
+import os, random, gc, re, json, time, shutil, glob
 import PIL.Image
 import tqdm
 from accelerate import Accelerator
@@ -14,19 +14,69 @@ DO_NOT_TRACK=1
 accelerator = Accelerator(cpu=True)
 InferenceClient=InferenceClient()
 
-models =[]
+models =[
+    "runwayml/stable-diffusion-v1-5",
+    "prompthero/openjourney-v4",
+    "CompVis/stable-diffusion-v1-4",
+    "stabilityai/stable-diffusion-2-1",
+    "stablediffusionapi/edge-of-realism",
+    "MirageML/fantasy-scene",
+    "wavymulder/lomo-diffusion",
+    "sd-dreambooth-library/fashion",
+    "DucHaiten/DucHaitenDreamWorld",
+    "VegaKH/Ultraskin",
+    "kandinsky-community/kandinsky-2-1",
+    "MirageML/lowpoly-cyberpunk",
+    "thehive/everyjourney-sdxl-0.9-finetuned",
+    "plasmo/woolitize-768sd1-5",
+    "plasmo/food-crit",
+    "johnslegers/epic-diffusion-v1.1",
+    "Fictiverse/ElRisitas",
+    "robotjung/SemiRealMix",
+    "herpritts/FFXIV-Style",
+    "prompthero/linkedin-diffusion",
+    "RayHell/popupBook-diffusion",
+    "MirageML/lowpoly-world",
+    "deadman44/SD_Photoreal_Merged_Models",
+    "johnslegers/epic-diffusion",
+    "tilake/China-Chic-illustration",
+    "wavymulder/modelshoot",
+    "prompthero/openjourney-lora",
+    "Fictiverse/Stable_Diffusion_VoxelArt_Model",
+    "darkstorm2150/Protogen_v2.2_Official_Release",
+    "hassanblend/HassanBlend1.5.1.2",
+    "hassanblend/hassanblend1.4",
+    "nitrosocke/redshift-diffusion",
+    "prompthero/openjourney-v2",
+    "nitrosocke/Arcane-Diffusion",
+    "Lykon/DreamShaper",
+    "wavymulder/Analog-Diffusion",
+    "nitrosocke/mo-di-diffusion",
+    "dreamlike-art/dreamlike-diffusion-1.0",
+    "dreamlike-art/dreamlike-photoreal-2.0",
+    "digiplay/RealismEngine_v1",
+    "digiplay/AIGEN_v1.4_diffusers",
+    "stablediffusionapi/dreamshaper-v6",
+    "p1atdev/liminal-space-diffusion",
+    "nadanainone/gigaschizonegs",
+    "lckidwell/album-cover-style",
+    "axolotron/ice-cream-animals",
+    "perion/ai-avatar",
+    "digiplay/GhostMix",
+    "ThePioneer/MISA",
+    "TheLastBen/froggy-style-v21-768",
+    "FloydianSound/Nixeu_Diffusion_v1-5",
+    "kakaobrain/karlo-v1-alpha-image-variations",
+    "digiplay/PotoPhotoRealism_v1",
+    "ConsistentFactor/Aurora-By_Consistent_Factor",
+    "rim0/quadruped_mechas",
+    "Akumetsu971/SD_Samurai_Anime_Model",
+    "Bojaxxx/Fantastic-Mr-Fox-Diffusion",
+    "sd-dreambooth-library/original-character-cyclps",
+]
 loris=[]
 apol=[]
 
-def hgfdm(models):
-    models=models
-    poi=InferenceClient.list_deployed_models()
-    voi=poi["text-to-image"]
-    for met in voi:
-        pio=""+met+""
-        models.append(pio)
-    return models
-
 def smdls(models):
     models=models
     mtlst=HfApi.list_models(filter="diffusers:StableDiffusionPipeline",limit=500,full=True,)
@@ -50,7 +100,7 @@ def chdr(apol,prompt,modil,los,stips,fnamo,gaul):
         type="SD"
         tre='./tmpo/'+fnamo+'.json'
         tra='./tmpo/'+fnamo+'.png'
-        flng=["yssup", "sllab", "stsaerb", "sinep", "selppin", "ssa", "tnuc", "mub", "kcoc", "kcid", "anigav", "dekan", "edun", "slatineg", "xes", "nrop", "stit", "ttub", "bojwolb", "noitartenep", "kcuf", "kcus", "kcil",]
+        flng=["yssup", "sllab", "stsaerb", "sinep", "selppin", "ssa", "tnuc", "mub", "kcoc", "kcid", "anigav", "dekan", "edun", "slatineg", "xes", "nrop", "stit", "ttub", "bojwolb", "noitartenep", "kcuf", "kcus", "kcil", "elttil", "gnuoy", "thgit", "lrig", "etitep", "dlihc", "yxes"]
         flng=[itm[::-1] for itm in flng]
         ptn = r"\b" + r"\b|\b".join(flng) + r"\b"
         if re.search(ptn, prompt, re.IGNORECASE):
@@ -64,7 +114,11 @@ def chdr(apol,prompt,modil,los,stips,fnamo,gaul):
         with open(tre, 'w') as f:
             json.dump(dobj, f)
         HfApi.upload_folder(repo_id="JoPmt/Tst_datast_imgs",folder_path="./tmpo",repo_type="dataset",path_in_repo="./",token=HF_TOKEN)
-        try: 
+        try:
+            for pgn in glob.glob('./tmpo/*.png'):
+                os.remove(pgn)
+            for jgn in glob.glob('./tmpo/*.json'):
+                os.remove(jgn)
             del tre
             del tra
         except: