File size: 422 Bytes
66ad25b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import os
import json
import requests
from dotenv import load_dotenv
load_dotenv()
key = os.getenv("NVIDIA_API_KEY_RERANK")
headers = {"Authorization": f"Bearer {key}"}
try:
r = requests.get("https://integrate.api.nvidia.com/v1/models", headers=headers)
models = r.json()
with open("nvidia_models.json", "w", encoding="utf-8") as f:
json.dump(models, f, indent=2)
except Exception as e:
print(e)
|