Spaces:

IndexTeam
/

Index-1.9B-Character

Runtime error

李新豪 commited on Jun 19, 2024

Commit

f420f35

1 Parent(s): 9462146

refresh

Files changed (1) hide show

src/retrieve_dialog.py CHANGED Viewed

@@ -33,7 +33,7 @@ class RetrieveDialog:
         # 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)

         # if not os.path.exists(local_dir):
         #     print("Please download bge-large-zh-v1.5 first!")
+        self.emb_model = SentenceTransformer("BAAI/bge-large-zh-v1.5")
         self.dialogs, self.context_index = self._get_emb_base_by_list(raw_dialog_list)