From 2412c26342bbd7e3c1462e47d337c5305723a2ab Mon Sep 17 00:00:00 2001
From: zhanghuan <1262329256@qq.com>
Date: Wed, 10 Sep 2025 15:12:45 +0800
Subject: [PATCH] =?UTF-8?q?settings.py=E4=B8=AD=E5=A2=9E=E5=8A=A0=E6=8E=A8?=
 =?UTF-8?q?=E7=90=86=E6=A8=A1=E5=9E=8B=E8=B7=AF=E5=BE=84=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 classifier/food_classifier_app.py | 3 ++-
 settings/settings.py              | 3 +++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/classifier/food_classifier_app.py b/classifier/food_classifier_app.py
index ff46079..277f248 100644
--- a/classifier/food_classifier_app.py
+++ b/classifier/food_classifier_app.py
@@ -14,6 +14,7 @@ from tkinterdnd2 import DND_FILES, TkinterDnD
 import threading
 import time
 from net import create_food_cnn
+from settings import settings
 
 # 设置customtkinter的外观
 ctk.set_appearance_mode("System")
@@ -54,7 +55,7 @@ class FoodClassifierApp:
     def load_model(self):
         """加载训练好的PyTorch模型"""
         try:
-            model_path = "../model/02/best_food_model.pth"
+            model_path = settings.INFERENCE_BEST_MODEL_PATH
             if os.path.exists(model_path):
                 # 创建模型实例
                 self.model = create_food_cnn()
diff --git a/settings/settings.py b/settings/settings.py
index e5427f0..8e9337c 100644
--- a/settings/settings.py
+++ b/settings/settings.py
@@ -19,6 +19,9 @@ MODEL_DIR = os.path.join(BASE_DIR, 'model', '03')
 BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth')
 TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png')
 
+# INFERENCE_BEST_MODEL_PATH = os.path.join(BASE_DIR, 'model', '03','best_food_model.pth')
+INFERENCE_BEST_MODEL_PATH = BEST_MODEL_PATH
+
 # 训练参数
 # NUM_EPOCHS = 100
 NUM_EPOCHS = 3