self-tools
/
bert_cate
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

1

This commit is contained in:
renzhiyuan 2025-08-11 15:18:00 +08:00
parent 3a7515ccc0
commit 9ea2e272ae
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app.py
View File

@ -11,8 +11,8 @@ import threading
# 配置参数 # 配置参数
MAX_LENGTH = 512 MAX_LENGTH = 512
DEVICE = "cuda" if torch.cuda.is_available() else "cpu" DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
SAVE_DIR = "./cate1" # 模型路径 SAVE_DIR = "./brand" # 模型路径
LABEL_ENCODER_PATH = "cate1/label_encoder.pkl" # 标签映射器路径 LABEL_ENCODER_PATH = "brand/label_encoder.pkl" # 标签映射器路径
app = Flask(__name__) app = Flask(__name__)
app.config['JSON_AS_ASCII'] = False app.config['JSON_AS_ASCII'] = False