Update app.py

app.py

@@ -270,7 +270,15 @@ PENDING_LORA_KEY: str | None = None
 PENDING_LORA_STATE: dict[str, torch.Tensor] | None = None
 PENDING_LORA_STATUS: str = "No LoRA state prepared yet."
 
-checkpoint_path = hf_hub_download(repo_id="SulphurAI/Sulphur-2-base", filename="sulphur_distil_bf16.safetensors")
+weights_dir = Path("weights")
+weights_dir.mkdir(exist_ok=True)
+checkpoint_path = hf_hub_download(
+    repo_id="SulphurAI/Sulphur-2-base",
+    filename="sulphur_distil_bf16.safetensors",
+    local_dir=str(weights_dir),
+    local_dir_use_symlinks=False,
+)
+
 spatial_upsampler_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename="ltx-2.3-spatial-upscaler-x2-1.1.safetensors")
 gemma_root = snapshot_download(repo_id=GEMMA_REPO)