35 lines
919 B
Python
35 lines
919 B
Python
|
"""
|
||
|
|
Script for running the model using the Hugging Face endpoint. An authorized Hugging Face API key is required.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import requests
|
||
|
|
import os
|
||
|
|
|
||
|
|
API_URL = "https://otaf5w2ge8huxngl.eu-west-1.aws.endpoints.huggingface.cloud"
|
||
|
|
# Set your Hugging Face API key as an environment variable
|
||
|
|
api_key = os.environ.get("HF_API_KEY")
|
||
|
|
headers = {
|
||
|
|
"Accept": "application/json",
|
||
|
|
"Authorization": f"Bearer {api_key}",
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def query(payload):
|
||
|
|
response = requests.post(API_URL, headers=headers, json=payload)
|
||
|
|
return response.json()
|
||
|
|
|
||
|
|
|
||
|
|
output = query(
|
||
|
|
{
|
||
|
|
"inputs": "", # Can be left empty.
|
||
|
|
"input_a": "<text A>", # Required for all tasks.
|
||
|
|
"input_b": "<text B>", # Required for task 2 but not for task 1 or 3.
|
||
|
|
"task": 1 | 2 | 3, # Choose the task number.
|
||
|
|
"parameters": {
|
||
|
|
# Can be left empty
|
||
|
|
},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
print(output)
|