From 98351dc096749b03e2e89a739c864d9e1915d7ca Mon Sep 17 00:00:00 2001
From: ClovertaTheTrilobita <cloverta@petalmail.com>
Date: Mon, 22 Sep 2025 17:47:16 +0000
Subject: [PATCH] =?UTF-8?q?feature(config):=20=E6=96=B0=E5=A2=9E=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E8=87=AA=E5=AE=9A=E4=B9=89=E8=B7=AF=E5=BE=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/config.yaml |  1 +
 backend/cp_run.py   | 11 ++++++++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/backend/config.yaml b/backend/config.yaml
index 19b5208..3892524 100644
--- a/backend/config.yaml
+++ b/backend/config.yaml
@@ -1,4 +1,5 @@
 model:
+  save_dir: models
 
 data:
   root_dir: .
diff --git a/backend/cp_run.py b/backend/cp_run.py
index 37b40f6..68e9b91 100644
--- a/backend/cp_run.py
+++ b/backend/cp_run.py
@@ -1,8 +1,7 @@
-from cellpose import models, plot
-from cellpose.io import imread, save_masks
+import os
 from PIL import Image
 import numpy as np
-import os, datetime
+import datetime
 import time
 from omegaconf import OmegaConf
 from pathlib import Path
@@ -14,6 +13,12 @@ BASE_DIR = cfg.data.root_dir
 UPLOAD_DIR = cfg.data.upload_dir
 OUTPUT_DIR = cfg.data.run.output_dir
 OUTPUT_TEST_DIR = cfg.data.run.test_output_dir
+MODELS_DIR = str((CONFIG_PATH.parent / cfg.model.save_dir).resolve())
+os.makedirs(MODELS_DIR, exist_ok=True)
+os.environ["CELLPOSE_LOCAL_MODELS_PATH"] = MODELS_DIR
+
+from cellpose import models, plot
+from cellpose.io import imread, save_masks
 
 class Cprun: