app_config.py

"""Configuration constants and module-level side effects (weight downloads)."""

import os

from huggingface_hub import snapshot_download

WEIGHTS_DIR = snapshot_download(
    repo_id="ProteinDesignLab/caliby-weights", repo_type="model", token=os.environ.get("HF_TOKEN")
)

# Set MODEL_PARAMS_DIR so caliby's weight utilities can find/download files.
os.environ.setdefault("MODEL_PARAMS_DIR", WEIGHTS_DIR)