chore: 允许 ai-service Python 文件入库(.gitignore 例外)
This commit is contained in:
@@ -0,0 +1,10 @@
|
||||
import os
|
||||
|
||||
# 运行模式:mock(默认,无模型文件)| onnx(加载 best.onnx 真实推理)
|
||||
MODEL_MODE = os.getenv("MODEL_MODE", "mock")
|
||||
MODEL_PATH = os.getenv("MODEL_PATH", "models/best.onnx")
|
||||
# YOLO 类别(二分类训练基线:healthy/sick;7 类病种扩展后再调整)
|
||||
MODEL_LABELS = os.getenv("MODEL_LABELS", "healthy,sick").split(",")
|
||||
# mock 模式返回的固定结果
|
||||
MOCK_CLASS = os.getenv("MOCK_CLASS", "healthy")
|
||||
MOCK_CONFIDENCE = float(os.getenv("MOCK_CONFIDENCE", "0.95"))
|
||||
Reference in New Issue
Block a user