from huggingface_hub import snapshot_download
import os

HF_REPO = os.getenv("HF_REPO_ID", "DarkSting/tea_backend")
HF_TOKEN = os.getenv("HF_TOKEN", None)  # needed if repo is private

print(f"Downloading files from {HF_REPO}...")

snapshot_download(
    repo_id=HF_REPO,
    repo_type="model",        # or "dataset" depending on where you uploaded
    local_dir="/app",         # downloads into /app matching your structure
)

print("Download complete.")