gpt j api

gpt-j-api?

API для взаимодействия с языковой моделью и вариантами GPT-J! Вы можете использовать и тестировать модель двумя разными способами:

• Веб-приложение Streamlit по адресу http://api.vicgalle.net:8000/.
• Правильный API, описанный по адресу http://api.vicgalle.net:5000/docs.

Конечные точки открытого API?

Это конечные точки общедоступного API, которые не требуют аутентификации. Нажмите на каждый, чтобы увидеть параметры!

Генерация текста GPT-J?

• генерировать: POST /generate/

Классификация текста с нулевым выстрелом (многоязычная) ?

• классифицировать: POST /classify/

Использование API

• Питон:

 import requests
context = "In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English."
payload = {
    "context" : context ,
    "token_max_length" : 512 ,
    "temperature" : 1.0 ,
    "top_p" : 0.9 ,
}
response = requests . post ( "http://api.vicgalle.net:5000/generate" , params = payload ). json ()
print ( response )

• Python (классификация с нулевым выстрелом):

 import requests
payload = { 
    "sequence" : "The movie started slow, but in the end was absolutely amazing!" , 
    "labels" : "positive,neutral,negative" }
response = requests . post ( "http://api.vicgalle.net:5000/classify" , params = payload ). json ()
print ( response )

• Баш:

curl -X ' POST ' 
  ' http://api.vicgalle.net:5000/generate?context=In%20a%20shocking%20finding%2C%20scientists%20discovered%20a%20herd%20of%20unicorns%20living%20in%20a%20remote%2C%20previously%20unexplored%20valley%2C%20in%20the%20Andes%20Mountains.%20Even%20more%20surprising%20to%20the%20researchers%20was%20the%20fact%20that%20the%20unicorns%20spoke%20perfect%20English.&token_max_length=512&temperature=1&top_p=0.9 ' 
  -H ' accept: application/json ' 
  -d ' ' 

Развертывание API-сервера

Просто подключите SSH к виртуальной машине TPU. Этот код был протестирован как на вариантах v2-8, так и на v3-8.

Сначала установим требования и получим веса:

 python3 -m pip install -r requirements.txt
wget https://the-eye.eu/public/AI/GPT-J-6B/step_383500_slim.tar.zstd
sudo apt install zstd
tar -I zstd -xf step_383500_slim.tar.zstd
rm step_383500_slim.tar.zstd

И просто беги

 python3 serve.py

Затем вы можете перейти по адресу http://localhost:5000/docs и использовать API!

Развертывание панели управления с подсветкой

Просто беги

 python3 -m streamlit run streamlit_app.py --server.port 8000

Контакт

Если у вас есть запрос, я буду рад вам помочь: vgallegoalcala at gmail dot com

Спонсоры?

Особая благодарность следующим людям, спонсирующим этот проект! <3

• Аспи96

Благодарности

Большое спасибо за поддержку Исследовательского облака ТПУ, https://sites.research.google/trc/