fix: paths

app.py

@@ -1,16 +1,33 @@
 import os
 import gradio as gr
 from scepter.modules.utils.file_system import FS
+from huggingface_hub import hf_hub_download
+
+def resolve_hf_path(path):
+    if isinstance(path, str) and path.startswith("hf://"):
+        # The expected format is: hf://{repo_id}@{filename}
+        parts = path[len("hf://"):].split("@")
+        if len(parts) != 2:
+            raise ValueError(f"Invalid HF URI format: {path}")
+        repo_id = parts[0]
+        filename = parts[1]
+        print(f"Downloading {filename} from {repo_id} ...")
+        local_path = hf_hub_download(repo_id=repo_id, filename=filename)
+        return local_path
+    return path
 
-# Set necessary environment variables for ACE++
 os.environ["FLUX_FILL_PATH"] = "hf://black-forest-labs/FLUX.1-Fill-dev"
 os.environ["PORTRAIT_MODEL_PATH"] = "ms://iic/ACE_Plus@portrait/comfyui_portrait_lora64.safetensors"
 os.environ["SUBJECT_MODEL_PATH"] = "ms://iic/ACE_Plus@subject/comfyui_subject_lora16.safetensors"
 os.environ["LOCAL_MODEL_PATH"] = "ms://iic/ACE_Plus@local_editing/comfyui_local_lora16.safetensors"
 os.environ["ACE_PLUS_FFT_MODEL"] = "hf://ali-vilab/ACE_Plus@ace_plus_fft.safetensors"
 
-os.environ["FLUX_FILL_PATH"]=FS.get_dir_to_local_dir(os.environ["FLUX_FILL_PATH"])
-os.environ["ACE_PLUS_FFT_MODEL"]=FS.get_from(os.environ["ACE_PLUS_FFT_MODEL"])
+flux_fill_path = resolve_hf_path(os.environ["FLUX_FILL_PATH"])
+ace_plus_fft_model_path = resolve_hf_path(os.environ["ACE_PLUS_FFT_MODEL"])
+
+# Update the environment variables with the resolved local file paths.
+os.environ["FLUX_FILL_PATH"] = flux_fill_path
+os.environ["ACE_PLUS_FFT_MODEL"] = ace_plus_fft_model_path
 
 from inference.ace_plus_inference import ACEInference
 from scepter.modules.utils.config import Config

requirements.txt

@@ -4,4 +4,6 @@ scepter
 torch
 torchvision
 transformers
-lightning
+lightning
+huggingface_hub
+safetensors