import sys, os
from huggingface_hub import HfApi

token = os.environ["HF_TOKEN"]
api = HfApi(token=token)

models = [
    "pyannote/speaker-diarization-3.1",
    "pyannote/segmentation-3.0",
]

all_ok = True
for m in models:
    try:
        info = api.model_info(m)
        print(f"PASS: {m} (sha={info.sha[:8]})")
    except Exception as e:
        all_ok = False
        print(f"FAIL: {m} -> {e}")

sys.exit(0 if all_ok else 1)