EngineX-Iluvatar/enginex-bi_150-text-classification
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases 2 Wiki Activity

init project

Browse Source
This commit is contained in:
aiyueqi
2026-04-10 10:05:19 +00:00
parent 849078e193
commit 5a21e8d6a2
7 changed files with 207 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
README.md
View File

@@ -1,2 +1,21 @@
# enginex-bi_150-text-classification
## Quickstart
### 启动服务
修改docker.sh的脚本中$mountpath为本地的模型挂载路径
然后运行./docker.sh
当打印出以下内容时表示模型load成功
```
2026-04-10 10:01:00 /workspace/transformers_server.py INFO model loaded successfully
INFO: Application startup complete.
INFO: Uvicorn running on http://0.0.0.0:8000 (Press CTRL+C to quit)
```
### 运行测试
在容器中执行 python3 test.py
打印出以下内容, 文本分类结果的id和lable:
```
text: 这家餐厅的服务特别周到，菜也很好吃，我下次还会再来。
predicted_class_id: 4
predicted_label: 5 stars
```