File size: 465 Bytes
c9b0079 |
1 2 3 4 5 6 7 8 9 10 11 |
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) |