diff --git a/.env b/.env index e3bb2ca..ef6f854 100644 --- a/.env +++ b/.env @@ -6,17 +6,17 @@ MODEL_CACHE_DIR=/app/model_cache # ==================== 税务分类模型 ==================== TAX_MODEL_ID=circles1/tax_rate TAX_MODEL_DIR=/app/services/tax/model -TAX_PORT=5003 +TAX_PORT=5004 TAX_SERVICE_NAME=tax_classifier # ==================== 订单地址模型 ==================== ADDRESS_MODEL_ID=circles1/order_address ADDRESS_MODEL_DIR=/app/services/address/model -ADDRESS_PORT=5002 +ADDRESS_PORT=5003 ADDRESS_SERVICE_NAME=address_classifier # ==================== 运行配置 ==================== DEVICE=cuda MAX_WORKERS=4 -BATCH_SIZE=32 +BATCH_SIZE=4 diff --git a/base/requirements.txt b/base/requirements.txt index a3ae07a..89010c0 100644 --- a/base/requirements.txt +++ b/base/requirements.txt @@ -6,4 +6,4 @@ gunicorn==26.0.0 requests==2.34.2 modelscope==1.39.1 load_dotenv==0.1.0 -scikit-learn \ No newline at end of file +scikit-learn==1.7.1 \ No newline at end of file