readinglevelconverter / arte_score.py
HakshaySundar's picture
Adding files for api integration
2610aed verified
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")