Update app.py

app.py

@@ -30,12 +30,25 @@ import gradio as gr
 import torch
 from worldgen import WorldGen
 
-# Lazy init to avoid long cold start
-_wg = None
+from huggingface_hub import hf_hub_download
+
 def get_worldgen():
     global _wg
     if _wg is None:
-        _wg = WorldGen(mode="t2s", device="cpu", low_vram=True, lora_path=None)
+        # 確保使用公開模型且不帶 token
+        model_path = hf_hub_download(
+            repo_id="nunchaku-tech/nunchaku-flux.1-dev",
+            filename="unquantized_layers.safetensors",
+            token=None  # 強制匿名模式
+        )
+
+        # 初始化 WorldGen，並指定模型路徑（如果需要）
+        _wg = WorldGen(
+            mode="t2s",
+            device="cpu",
+            low_vram=True,
+            lora_path=None  # 如果你有 LoRA 模型可以指定路徑
+        )
     return _wg
 
 def _zip_directory(dir_path: Path) -> str: