Update app.py

app.py

@@ -1,13 +1,9 @@
-import os
-os.environ['CUDA_VISIBLE_DEVICES'] = '3'
-
 import os
 import sys
 import faiss
 import numpy as np
 import streamlit as st
 from text2vec import SentenceModel
-# 请确保 JSONLIndexer 在 src 目录下或者已正确安装
 from src.jsonl_Indexer import JSONLIndexer
 
 # 命令行参数处理函数
@@ -27,7 +23,7 @@ cli_args = get_cli_args()
 # 设置默认值（适用于 JSONL 文件）
 DEFAULT_CONFIG = {
     'model_path': 'BAAI/bge-base-en-v1.5',
-    'dataset_path': 'src/tool-embedding.jsonl',  # JSONL 文件路径
+    'dataset_path': 'tool-embedding.jsonl',  # JSONL 文件路径
     'vector_size': 768,
     'embedding_field': 'embedding',   # JSON中存储embedding的字段名
     'id_field': 'id'                  # JSON中作为待检索文本的字段