import json
import requests
API_URL = "https://api-inference.huggingface.co/models/allenai/longformer-large-4096-finetuned-triviaqa"
headers = {"Authorization": f"Bearer {API_TOKEN}"}
defquery(payload):
data = json.dumps(payload)
response = requests.request("POST", API_URL, headers=headers, data=data)
return json.loads(response.content.decode("utf-8"))
data = query(
{
"inputs": {
"question": "What's my name?",
"context": "My name is Clara and I live in Berkeley.",
}
}
)