Upload app.py

app.py

@@ -1,31 +1,7 @@
 import gradio as gr
 from pathlib import Path
 
-models = [
-    'cagliostrolab/animagine-xl-3.1',
-    'votepurchase/ponyDiffusionV6XL',
-    'yodayo-ai/kivotos-xl-2.0',
-    'yodayo-ai/holodayo-xl-2.1',
-    'stabilityai/stable-diffusion-xl-base-1.0',
-    'eienmojiki/Anything-XL',
-    'eienmojiki/Starry-XL-v5.2',
-    'digiplay/majicMIX_sombre_v2',
-    'digiplay/majicMIX_realistic_v7',
-    'digiplay/DreamShaper_8',
-    'digiplay/BeautifulArt_v1',
-    'digiplay/DarkSushi2.5D_v1',
-    'digiplay/darkphoenix3D_v1.1',
-    'digiplay/BeenYouLiteL11_diffusers',
-    'votepurchase/counterfeitV30_v30',
-    'Meina/MeinaMix_V11',
-    'Meina/MeinaUnreal_V5',
-    'Meina/MeinaPastel_V7',
-    'KBlueLeaf/Kohaku-XL-Epsilon-rev2',
-    'KBlueLeaf/Kohaku-XL-Epsilon-rev3',
-    'kayfahaarukku/UrangDiffusion-1.1',
-    'Raelina/Rae-Diffusion-XL-V2',
-    'Raelina/Raemu-XL-V4',
-]
+
 loaded_models = {}
 model_info_dict = {}
 
@@ -34,11 +10,38 @@ def list_sub(a, b):
     return [e for e in a if e not in b]
 
 
+def list_uniq(l):
+        return sorted(set(l), key=l.index)
+
+
 def is_repo_name(s):
     import re
     return re.fullmatch(r'^[^/]+?/[^/]+?$', s)
 
 
+def find_model_list(author: str="", tags: list[str]=[], not_tag="", sort: str="last_modified", limit: int=30):
+    from huggingface_hub import HfApi
+    api = HfApi()
+    default_tags = ["diffusers"]
+    models = []
+    try:
+        model_infos = api.list_models(author=author, task="text-to-image", pipeline_tag="text-to-image",
+                                       tags=list_uniq(default_tags + tags), cardData=True, sort=sort, limit=limit * 5)
+    except Exception as e:
+        print(f"Error: Failed to list models.")
+        print(e)
+        return models
+    for model in model_infos:
+        if not model.private and not model.gated:
+           if not_tag and not_tag in model.tags: continue
+           models.append(model.id)
+           if len(models) == limit: break
+    return models
+
+
+models = find_model_list("John6666", ["anime"])
+
+
 def get_t2i_model_info_dict(repo_id: str):
     from huggingface_hub import HfApi
     api = HfApi()
@@ -176,8 +179,6 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
         f"""This demo was created in reference to the following demos.
 - [Nymbo/Flood](https://huggingface.co/spaces/Nymbo/Flood).
 - [Yntec/ToyWorldXL](https://huggingface.co/spaces/Yntec/ToyWorldXL).
-<br>The first startup takes a mind-boggling amount of time, but not so much after the second.
-This is due to the time it takes for Gradio to generate an example image to cache.
             """
     )
     gr.DuplicateButton(value="Duplicate Space")