format code (#14)

2023-12-25 10:17:13 +08:00
parent 8a65ad888e
commit 6858ec5f38
30 changed files with 1062 additions and 771 deletions
--- a/python/svr/parse_user_docs.py
+++ b/python/svr/parse_user_docs.py
@@ -209,10 +209,8 @@ def rm_doc_from_kb(df):

 def main(comm, mod):
    global model
-    from FlagEmbedding import FlagModel
-    model = FlagModel('/opt/home/kevinhu/data/bge-large-zh-v1.5/',
-                      query_instruction_for_retrieval="为这个句子生成表示以用于检索相关文章：",
-                      use_fp16=torch.cuda.is_available())
+    from llm import HuEmbedding
+    model = HuEmbedding()
    tm_fnm = f"res/{comm}-{mod}.tm"
    tm = findMaxDt(tm_fnm)
    rows = collect(comm, mod, tm)