text-analytics / oneai.py
afiz's picture
added api_key
a38d269
# Edit this One AI API call using our studio at https://studio.oneai.com/?pipeline=j2Yiyk
import os
import requests
api_key = os.environ['api_key']
url = "https://api.oneai.com/api/v0/pipeline"
def oneai_pipeline(text):
headers = {
"api-key": api_key,
"content-type": "application/json"
}
payload = {
"input": text,
"input_type": "article",
"steps": [
{
"skill": "summarize",
"params": {
"auto_length": True,
"max_length": 100,
"min_length": 5
}
},
]
}
r = requests.post(url, json=payload, headers=headers)
data = r.json()
return data['output'][0]['text']