增加一些设置。

2025-10-17 18:24:04 +08:00
parent 25f8e0a267
commit 5db96600fe
10 changed files with 38 additions and 25 deletions
@@ -87,7 +87,7 @@ def run_search_demo():
        from faiss_vector_db.build_faiss_index import FAISSSearcher
        
        # 配置参数
-        index_dir = "faiss_index"
+        index_dir = "DishClassification/faiss_index"
        model_path = "../model/embedding_20250917_145342/best_embedding_model.pth"
        
        # 检查索引是否存在
@@ -206,7 +206,7 @@ def main():
    print("✓ 文件检查通过")
    
    # 检查是否已有索引
-    index_exists = os.path.exists("faiss_index/faiss_index.bin")
+    index_exists = os.path.exists("DishClassification/faiss_index/faiss_index.bin")
    
    if index_exists:
        print("✓ 发现已存在的FAISS索引")