Baseline Hf API

1#!/usr/bin/env python3
2"""
3Ultra-simple Hugging Face API example (Python).
4
5Fetches a small list of models from the HF API and prints raw JSON.
6Uses HF_TOKEN for auth if the environment variable is set.
7"""
8
9from __future__ import annotations
10
11import os
12import sys
13import urllib.request
14
15
16def show_help() -> None:
17    print(
18        """Ultra-simple Hugging Face API example (Python)
19
20Usage:
21  baseline_hf_api.py [limit]
22  baseline_hf_api.py --help
23
24Description:
25  Fetches a small list of models from the HF API and prints raw JSON.
26  Uses HF_TOKEN for auth if the environment variable is set.
27
28Examples:
29  baseline_hf_api.py
30  baseline_hf_api.py 5
31  HF_TOKEN=your_token baseline_hf_api.py 10
32"""
33    )
34
35
36def main() -> int:
37    if len(sys.argv) > 1 and sys.argv[1] == "--help":
38        show_help()
39        return 0
40
41    limit = sys.argv[1] if len(sys.argv) > 1 else "3"
42    if not limit.isdigit():
43        print("Error: limit must be a number", file=sys.stderr)
44        return 1
45
46    token = os.getenv("HF_TOKEN")
47    headers = {"Authorization": f"Bearer {token}"} if token else {}
48    url = f"https://huggingface.co/api/models?limit={limit}"
49
50    req = urllib.request.Request(url, headers=headers)
51    with urllib.request.urlopen(req) as resp:
52        sys.stdout.write(resp.read().decode("utf-8"))
53    return 0
54
55
56if __name__ == "__main__":
57    raise SystemExit(main())