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