Auto-download weights from LogicGoInfotechSpaces/Smile_Changer_pre_model if missing

app.py

@@ -4,6 +4,7 @@ import logging
 from typing import Tuple, Dict
 
 import gradio as gr
+from huggingface_hub import snapshot_download
 from PIL import Image
 
 from runners.simple_runner import SimpleRunner
@@ -22,9 +23,29 @@ logger = logging.getLogger("sfe-app")
 RUNNER: SimpleRunner | None = None
 
 
+def ensure_weights():
+    """Make sure pretrained weights exist locally; otherwise fetch from your HF model repo."""
+    need = [
+        "pretrained_models/sfe_editor_light.pt",
+        "pretrained_models/stylegan2-ffhq-config-f.pt",
+    ]
+    if all(os.path.exists(p) for p in need):
+        return
+
+    repo_id = "LogicGoInfotechSpaces/Smile_Changer_pre_model"
+    logger.info("Missing weights; downloading snapshot from %s", repo_id)
+    snapshot_download(
+        repo_id=repo_id,
+        local_dir="pretrained_models",
+        local_dir_use_symlinks=False,
+        allow_patterns=["**/*"],
+    )
+
+
 def get_runner() -> SimpleRunner:
     global RUNNER
     if RUNNER is None:
+        ensure_weights()
         logger.info("Initializing SimpleRunner with %s", "pretrained_models/sfe_editor_light.pt")
         RUNNER = SimpleRunner(
             editor_ckpt_pth="pretrained_models/sfe_editor_light.pt",