| import requests | |
| def query(payload, model_id, api_token): | |
| headers = {"Authorization": f"Bearer {api_token}"} | |
| API_URL = f"https://api-inference.huggingface.co/models/{model_id}" | |
| response = requests.post(API_URL, headers=headers, json=payload) | |
| return response.json() | |
| model_id = "distilbert-base-uncased" | |
| api_token = "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" # get yours at hf.co/settings/tokens | |
| data = query("The goal of life is [MASK].", model_id, api_token) |