refresh

__init__.py

@@ -1,6 +0,0 @@
-# coding=utf-8
-from .get_dataset import *
-from .logger import *
-from .prompt_concat import *
-from .retrieve_dialog import *
-from .utils import *

src/retrieve_dialog.py

@@ -29,11 +29,11 @@ class RetrieveDialog:
 
         # config = load_json("config/config.json")
         # local_dir = config["bge_local_path"]
-        local_dir = os.environ.get('MODEL_PATH', 'IndexTeam/Index-1.9B-Character')
+        # local_dir = os.environ.get('MODEL_PATH', 'BAAI/bge-large-zh-v1.5')
 
-        if not os.path.exists(local_dir):
-            print("Please download bge-large-zh-v1.5 first!")
-        self.emb_model = SentenceTransformer(local_dir)
+        # if not os.path.exists(local_dir):
+        #     print("Please download bge-large-zh-v1.5 first!")
+        self.emb_model = SentenceTransformer("bge-large-zh-v1.5")
 
         self.dialogs, self.context_index = self._get_emb_base_by_list(raw_dialog_list)