15 lines
526 B
Python
15 lines
526 B
Python
import requests
|
|
|
|
API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-mnli"
|
|
headers = {"Authorization": "Bearer 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"}
|
|
|
|
def query(payload):
|
|
response = requests.post(API_URL, headers=headers, json=payload)
|
|
return response.json()
|
|
|
|
output = query({
|
|
"inputs": "Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!",
|
|
"parameters": {"candidate_labels": ["refund", "legal", "faq"]},
|
|
})
|
|
|
|
print(output) |