Spaces:
Sleeping
Sleeping
| import requests | |
| import json | |
| import os | |
| ARTE_API_KEY = os.environ.get("ARTE_API_KEY") | |
| ARTE_BASE_URL = os.environ.get("ARTE_BASE_URL") | |
| arte_api_endpoint="ping_api" | |
| def ping_api(text, model_type="SBERT"): | |
| payload = { | |
| "APIkey": ARTE_API_KEY, # Replace with your actual API key | |
| "selection": [model_type], | |
| "text": text | |
| } | |
| json_payload = json.dumps(payload) | |
| headers = { | |
| 'Content-Type': 'application/json' | |
| } | |
| response = requests.post(ARTE_BASE_URL + arte_api_endpoint, data=json_payload, headers=headers) | |
| data = json.loads(response.text) | |
| return data.get(model_type).get("sc") |