import os
import requests

API_URL = os.environ['HF_API_URL']
headers = {
    "Accept": "application/json",
    "Authorization": f"Bearer {os.environ['HF_API_KEY']}",
    "Content-Type": "application/json"
}


def query(payload):
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()


sentence = '''דוד בן-גוריון (16 באוקטובר 1886 - ו' בכסלו תשל"ד) היה מדינאי ישראלי וראש הממשלה הראשון של מדינת ישראל.'''

output = query({
    "inputs": sentence,
    "parameters": {
        "aggregation_strategy": "simple"
    }
})

print(output)