Upload folder using huggingface_hub

requirements.txt

@@ -1,52 +1,56 @@
-addict
+comfyui-frontend-package==1.33.10
+comfyui-workflow-templates==0.7.25
+comfyui-embedded-docs==0.3.1
+torch
+torchsde
+torchvision
+torchaudio
+numpy>=1.25.0
+einops
+transformers>=4.50.3
+tokenizers>=0.13.3
+sentencepiece
+safetensors>=0.4.2
 aiohttp>=3.11.8
+yarl>=1.18.0
+pyyaml
+Pillow
+scipy
+tqdm
+psutil
 alembic
-albumentations
+SQLAlchemy
 av>=14.2.0
-comfyui-embedded-docs
-comfyui-frontend-package
-comfyui-workflow-templates
-einops
+
+#non essential dependencies:
+kornia>=0.7.1
+spandrel
+pydantic~=2.0
+pydantic-settings~=2.0
+
+
+addict
+albumentations
 filelock
 ftfy
 fvcore
-gradio
 huggingface-hub
 imageio[ffmpeg]
 importlib_metadata
-kornia>=0.7.1
 matplotlib
 mediapipe
 ninja
-numpy>=1.25.0
 omegaconf
 opencv-python>=4.7.0.72
-Pillow
-psutil
-pydantic~=2.0
-pydantic-settings~=2.0
 python-dateutil
-PyYAML
 requests
-safetensors>=0.4.2
 scikit-image
 scikit-learn
-scipy
-sentencepiece
 soundfile
 spaces
-spandrel
-SQLAlchemy
 svglib
-tokenizers>=0.13.3
-torch
-torchaudio
 torchsde
-torchvision
-tqdm
-transformers>=4.37.2
 trimesh[easy]
 yacs
 yapf
-yarl>=1.18.0
 onnxruntime-gpu

ui/layout.py

@@ -24,8 +24,8 @@ def build_ui(event_handler_function):
         gr.Markdown("# ImageGen - NoobAI")
         gr.Markdown(
             "This demo is a streamlined version of the [Comfy web UI](https://github.com/RioShiina47/comfy-webui)'s ImgGen functionality. "
-            "Due to storage limitations on Spaces, this demo features the NoobAI series of models.\n"
             "Other versions are also available: "
+            "[Z-Image](https://huggingface.co/spaces/RioShiina/ImageGen-Z-Image), "
             "[Illstrious](https://huggingface.co/spaces/RioShiina/ImageGen-Illstrious), "
             "[Pony](https://huggingface.co/spaces/RioShiina/ImageGen-Pony1), "
             "[SDXL](https://huggingface.co/spaces/RioShiina/ImageGen-SDXL), "