...
This commit is contained in:
15
_archive/bart/bart.py
Normal file
15
_archive/bart/bart.py
Normal file
@@ -0,0 +1,15 @@
|
||||
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)
|
Reference in New Issue
Block a user