External/SnapOtter
1
0
Fork 0
mirror of https://github.com/snapotter-hq/SnapOtter.git synced 2026-08-03 07:46:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: use MODELS_PATH env var for AI model paths instead of hardcoded /opt/models

Browse Source 
The on-demand feature download system stores models at /data/ai/models/
(set via MODELS_PATH env var), but all Python scripts hardcoded
/opt/models/ as the base path. Each script now reads MODELS_PATH and
falls back to /opt/models for backward compatibility.
This commit is contained in:
ashim-hq
2026-04-18 10:17:01 +08:00
parent 28462c8f60
commit d7b6037b3d
9 changed files with 36 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/ai/python/colorize.py
+6 -4
View File
@@ -17,23 +17,25 @@ def emit_progress(percent, stage):
print(json.dumps({"progress": percent, "stage": stage}), file=sys.stderr, flush=True)
_MODELS_BASE = os.environ.get("MODELS_PATH", "/opt/models")
DDCOLOR_MODEL_PATH = os.environ.get(
"DDCOLOR_MODEL_PATH",
"/opt/models/ddcolor/ddcolor.onnx",
os.path.join(_MODELS_BASE, "ddcolor", "ddcolor.onnx"),
)
# OpenCV DNN fallback model paths (lightweight ~17 MB)
OPENCV_PROTO_PATH = os.environ.get(
"OPENCV_COLORIZE_PROTO",
"/opt/models/colorize-opencv/colorization_deploy_v2.prototxt",
os.path.join(_MODELS_BASE, "colorize-opencv", "colorization_deploy_v2.prototxt"),
)
OPENCV_MODEL_PATH = os.environ.get(
"OPENCV_COLORIZE_MODEL",
"/opt/models/colorize-opencv/colorization_release_v2.caffemodel",
os.path.join(_MODELS_BASE, "colorize-opencv", "colorization_release_v2.caffemodel"),
)
OPENCV_POINTS_PATH = os.environ.get(
"OPENCV_COLORIZE_POINTS",
"/opt/models/colorize-opencv/pts_in_hull.npy",
os.path.join(_MODELS_BASE, "colorize-opencv", "pts_in_hull.npy"),
)