moving to torch 2.5.1 for zerogpu

demo.py

@@ -5,7 +5,7 @@ if "HF_DEMO" in os.environ:
     # Global variable to track if install() has been run; only for deploying on HF space
     INSTALLED = False
     if not INSTALLED:
-        install()
+        install(is_hf_demo=True)
         INSTALLED = True
 
 import gradio as gr
@@ -148,7 +148,7 @@ def create_gradio_interface():
     demo.queue(max_size=99)
     
     try:
-        demo.launch()
+        demo.launch(share=True)
     except OSError as e:
         print(f"Error starting server: {e}")
         sys.exit(1)

face_enhance.py

@@ -3,7 +3,7 @@ import random
 import sys
 from typing import Sequence, Mapping, Any, Union
 import torch
-# import spaces
+import spaces
 COMFYUI_PATH = "./ComfyUI"
 
 """
@@ -313,7 +313,7 @@ def save_comfy_images(images, output_dirs):
         pil_image = Image.fromarray(numpy_image)
         pil_image.save(output_dirs[idx])
 
-# @spaces.GPU
+@spaces.GPU
 def face_enhance(face_image: str, input_image: str, output_image: str, dist_image: str = None, positive_prompt: str = "", id_weight: float = 0.75):
     initialize_models()  # Ensure models are loaded
     main(face_image, input_image, output_image, dist_image, positive_prompt, id_weight)

install.py

@@ -202,12 +202,19 @@ def install_custom_nodes():
     print("✅ Installed and updated all ComfyUI nodes.")
 
 
-def install():
+def install_hfdemo_dependencies():
+    run_cmd("python -m pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124")
+    run_cmd("python -m pip install -r requirements.txt")
+
+
+def install(is_hf_demo=False):
     install_lfs_files()
     install_comfyui()
     install_custom_nodes()
     download_huggingface_models()
     download_and_extract_antelopev2()
+    if is_hf_demo:
+        install_hfdemo_dependencies()
     print("🎉 Setup Complete!")
 
 if __name__ == "__main__":

requirements.txt

@@ -1,176 +1,3 @@
-aiofiles==24.1.0
-aiohappyeyeballs==2.6.1
-aiohttp==3.11.16
-aiosignal==1.3.2
-albucore==0.0.23
-albumentations==2.0.5
-annotated-types==0.7.0
-anyio==4.9.0
-arrow==1.3.0
-attrs==25.3.0
-av==14.3.0
-binaryornot==0.4.4
-certifi==2025.1.31
-cffi==1.17.1
-chardet==5.2.0
-charset-normalizer==3.4.1
-click==8.1.8
-coloredlogs==15.0.1
-comfy-cli==1.3.8
-comfyui_frontend_package==1.15.13
-contourpy==1.3.2
-cookiecutter==2.6.0
-cryptography==44.0.2
-cycler==0.12.1
-Cython==3.0.12
-Deprecated==1.2.18
-distro==1.9.0
-dnspython==2.7.0
-easydict==1.13
-einops==0.8.1
-email_validator==2.2.0
-facexlib==0.3.0
-fal_client==0.5.9
-fastapi==0.115.12
-fastapi-cli==0.0.7
-ffmpy==0.5.0
-filelock==3.18.0
-filterpy==1.4.5
-flatbuffers==25.2.10
-fonttools==4.57.0
-frozenlist==1.5.0
-fsspec==2025.3.2
-ftfy==6.3.1
-gitdb==4.0.12
-GitPython==3.1.44
 gradio==5.25.2
-gradio_client==1.8.0
-groovy==0.1.2
-h11==0.14.0
-hf_transfer==0.1.9
-httpcore==1.0.8
-httptools==0.6.4
-httpx==0.28.1
-httpx-sse==0.4.0
-huggingface-hub==0.30.2
-humanfriendly==10.0
-idna==3.10
-imageio==2.37.0
-insightface==0.7.3
-Jinja2==3.1.6
-jiter==0.9.0
-joblib==1.4.2
-kiwisolver==1.4.8
-kornia==0.8.0
-kornia_rs==0.1.8
-lazy_loader==0.4
-llvmlite==0.44.0
-markdown-it-py==3.0.0
-MarkupSafe==3.0.2
-matplotlib==3.10.1
-matrix-client==0.4.0
-mdurl==0.1.2
-mixpanel==4.10.1
-mpmath==1.3.0
-multidict==6.4.3
-networkx==3.4.2
-numba==0.61.2
-numpy==2.2.4
-nvidia-cublas-cu12==12.4.5.8
-nvidia-cuda-cupti-cu12==12.4.127
-nvidia-cuda-nvrtc-cu12==12.4.127
-nvidia-cuda-runtime-cu12==12.4.127
-nvidia-cudnn-cu12==9.1.0.70
-nvidia-cufft-cu12==11.2.1.3
-nvidia-curand-cu12==10.3.5.147
-nvidia-cusolver-cu12==11.6.1.9
-nvidia-cusparse-cu12==12.3.1.170
-nvidia-cusparselt-cu12==0.6.2
-nvidia-nccl-cu12==2.21.5
-nvidia-nvjitlink-cu12==12.4.127
-nvidia-nvtx-cu12==12.4.127
-onnx==1.17.0
-onnxruntime==1.21.0
-onnxruntime-gpu==1.21.0
-openai==1.75.0
-opencv-python==4.11.0.86
-opencv-python-headless==4.11.0.86
-orjson==3.10.16
-packaging==24.2
-pandas==2.2.3
-pathspec==0.12.1
-pillow==11.2.1
-prettytable==3.16.0
-prompt_toolkit==3.0.51
-propcache==0.3.1
-protobuf==6.30.2
-psutil==7.0.0
-pycparser==2.22
-pydantic==2.11.3
-pydantic_core==2.33.1
-pydub==0.25.1
-PyGithub==2.6.1
-Pygments==2.19.1
-PyJWT==2.10.1
-PyNaCl==1.5.0
-pyparsing==3.2.3
-python-dateutil==2.9.0.post0
-python-dotenv==1.1.0
-python-multipart==0.0.20
-python-slugify==8.0.4
-pytz==2025.2
-PyYAML==6.0.2
-questionary==2.1.0
-regex==2024.11.6
-requests==2.32.3
-rich==14.0.0
-rich-toolkit==0.14.1
-ruff==0.11.5
-safehttpx==0.1.6
-safetensors==0.5.3
-scikit-image==0.25.2
-scikit-learn==1.6.1
-scipy==1.15.2
-semantic-version==2.10.0
-semver==3.0.4
-sentencepiece==0.2.0
-shellingham==1.5.4
-simsimd==6.2.1
-six==1.17.0
-smmap==5.0.2
-sniffio==1.3.1
-soundfile==0.13.1
-spandrel==0.4.1
-starlette==0.46.2
-stringzilla==3.12.4
-sympy==1.13.1
-text-unidecode==1.3
-threadpoolctl==3.6.0
-tifffile==2025.3.30
-timm==1.0.15
-tokenizers==0.21.1
-toml==0.10.2
-tomlkit==0.13.2
-torch==2.6.0
-torchaudio==2.6.0
-torchsde==0.2.6
-torchvision==0.21.0
-tqdm==4.67.1
-trampoline==0.1.2
-transformers==4.51.3
-triton==3.2.0
-typer==0.15.2
-types-python-dateutil==2.9.0.20241206
-typing-inspection==0.4.0
-typing_extensions==4.13.2
-tzdata==2025.2
-urllib3==1.26.20
-uv==0.6.14
-uvicorn==0.34.1
-uvloop==0.21.0
-watchfiles==1.0.5
-wcwidth==0.2.13
-websocket-client==1.8.0
-websockets==15.0.1
-wrapt==1.17.2
-yarl==1.20.0
+pillow
+spaces