import requests

API_URL = "https://api-inference.huggingface.co/models/AdaptLLM/law-LLM"
API_TOKEN="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"
headers = {"Authorization": f"Bearer {API_TOKEN}"}


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


output = query({
    "inputs": "Can",
})
print(output)