| """ | |
| 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) | |