Spaces:
Running
on
Zero
Running
on
Zero
李新豪
commited on
Commit
•
f420f35
1
Parent(s):
9462146
refresh
Browse files- src/retrieve_dialog.py +1 -1
src/retrieve_dialog.py
CHANGED
|
@@ -33,7 +33,7 @@ class RetrieveDialog:
|
|
| 33 |
|
| 34 |
# if not os.path.exists(local_dir):
|
| 35 |
# print("Please download bge-large-zh-v1.5 first!")
|
| 36 |
-
self.emb_model = SentenceTransformer("bge-large-zh-v1.5")
|
| 37 |
|
| 38 |
self.dialogs, self.context_index = self._get_emb_base_by_list(raw_dialog_list)
|
| 39 |
|
|
|
|
| 33 |
|
| 34 |
# if not os.path.exists(local_dir):
|
| 35 |
# print("Please download bge-large-zh-v1.5 first!")
|
| 36 |
+
self.emb_model = SentenceTransformer("BAAI/bge-large-zh-v1.5")
|
| 37 |
|
| 38 |
self.dialogs, self.context_index = self._get_emb_base_by_list(raw_dialog_list)
|
| 39 |
|