Update app.py

app.py

@@ -252,7 +252,11 @@ def generate_storyboard_with_llm(user_prompt: str, n_shots: int, default_fps: in
 USE_CUDA = torch.cuda.is_available()
 DTYPE = torch.float16 if USE_CUDA else torch.float32
 
-FLUX_MODEL = os.getenv("FLUX_MODEL", "black-forest-labs/FLUX.1-Nano")
+# ✅ Use a real FLUX repo instead of Nano
+FLUX_MODEL = os.getenv("FLUX_MODEL", "black-forest-labs/FLUX.1-schnell")
+
+# ✅ Add token support
+HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
 
 _flux_t2i = None
 _flux_i2i = None
@@ -262,21 +266,23 @@ def _lazy_flux_pipes():
     global _flux_t2i, _flux_i2i
     if _flux_t2i is not None and _flux_i2i is not None:
         return _flux_t2i, _flux_i2i
-    _flux_t2i = FluxPipeline.from_pretrained(FLUX_MODEL, torch_dtype=DTYPE, use_safetensors=True)
+    _flux_t2i = FluxPipeline.from_pretrained(
+        FLUX_MODEL, torch_dtype=DTYPE, use_safetensors=True, token=HF_TOKEN
+    )
     if USE_CUDA: _flux_t2i = _flux_t2i.to("cuda")
-    _flux_i2i = FluxImg2ImgPipeline.from_pretrained(FLUX_MODEL, torch_dtype=DTYPE, use_safetensors=True)
+    _flux_i2i = FluxImg2ImgPipeline.from_pretrained(
+        FLUX_MODEL, torch_dtype=DTYPE, use_safetensors=True, token=HF_TOKEN
+    )
     if USE_CUDA: _flux_i2i = _flux_i2i.to("cuda")
     return _flux_t2i, _flux_i2i
 
 def _flux_healthcheck():
-    try:
-        _lazy_flux_pipes()
-        return True
-    except Exception as e:
+    if not HF_TOKEN:
         raise RuntimeError(
-            f"FLUX failed to initialize: {e}\n"
-            f"FLUX_MODEL='{FLUX_MODEL}'. If the repo is gated/private, set HF_TOKEN in env."
+            "HF_TOKEN is not set. FLUX models are gated; set a Hugging Face READ token "
+            "and accept the model terms on the repo page."
         )
+    _lazy_flux_pipes()
 
 def _save_keyframe(pid: str, shot_id: int, img: Image.Image) -> str:
     pdir = project_dir(pid)