跳转至

Google T5 翻译即服务,只需 7 行代码

什么是T5? Google 的 Text-To-Text Transfer Transformer (T5) 提供了翻译功能。

翻译

在本文中,我们将 Google T5 模型部署为 REST API 服务。 难的? 我告诉你怎么样:你只需要写 7 行代码?

翻译

安装依赖

HuggingFace

pip install "transformers[pytorch]"

如果不起作用,请访问 Installation 并查看其官方文档。

Pinferencia

pip install "pinferencia[uvicorn]"

定义服务

首先让我们创建 app.py 来定义服务:

app.py
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
from transformers import pipeline

from pinferencia import Server

t5 = pipeline(model="t5-base", tokenizer="t5-base")


def translate(text):
    return t5(text)


service = Server()
service.register(model_name="t5", model=translate)

启动服务

$ uvicorn app:service --reload
INFO:     Started server process [xxxxx]
INFO:     Waiting for application startup.
INFO:     Application startup complete.
INFO:     Uvicorn running on http://127.0.0.1:8000 (Press CTRL+C to quit)

测试服务

curl -X 'POST' \
    'http://localhost:8000/v1/models/t5/predict' \
    -H 'accept: application/json' \
    -H 'Content-Type: application/json' \
    -d '{
    "parameters": {},
    "data": "translate English to German: Good morning, my love."
}'

结果:

{
    "model_name": "t5",
    "data": [
        {
        "translation_text": "Guten Morgen, liebe Liebe."
        }
    ]
}
test.py
1
2
3
4
5
6
7
8
9
import requests

response = requests.post(
    url="http://localhost:8000/v1/models/gpt2/predict",
    json={
        "data": "translate English to German: Good morning, my love."
    },
)
print("Prediction:", response.json()["data"])

运行python test.py并打印结果:

Prediction: {
    "translation_text": "Guten Morgen, liebe Liebe."
}

更酷的是,访问 http://127.0.0.1:8000,您将拥有一个交互式 ui。

您可以在那里发送预测请求!