Bread-F commited on
Commit
bc00a21
·
verified ·
1 Parent(s): a49dbe4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -30,7 +30,7 @@ st.set_page_config(
30
  from utils.rag.rag_worker import gen_rag_db
31
  from utils.tools import resize_image
32
 
33
- from utils.model_loader import RAG_RETRIEVER # isort:skip
34
 
35
 
36
  @st.experimental_dialog("科室介绍", width="large")
@@ -495,13 +495,13 @@ def update_product_info(
495
  yaml.dump(product_info_dict, f, allow_unicode=True)
496
 
497
  st.write("生成数据库...")
498
- if WEB_CONFIGS.ENABLE_RAG:
499
- # 重新生成 RAG 向量数据库
500
- gen_rag_db(force_gen=True)
501
 
502
- # 重新加载 retriever
503
- RAG_RETRIEVER.pop("default")
504
- RAG_RETRIEVER.get(fs_id="default", config_path=WEB_CONFIGS.RAG_CONFIG_PATH, work_dir=WEB_CONFIGS.RAG_VECTOR_DB_DIR)
505
 
506
  # 更新状态
507
  status.update(label="添加科室信息成功!", state="complete", expanded=False)
 
30
  from utils.rag.rag_worker import gen_rag_db
31
  from utils.tools import resize_image
32
 
33
+ # from utils.model_loader import RAG_RETRIEVER # isort:skip
34
 
35
 
36
  @st.experimental_dialog("科室介绍", width="large")
 
495
  yaml.dump(product_info_dict, f, allow_unicode=True)
496
 
497
  st.write("生成数据库...")
498
+ # if WEB_CONFIGS.ENABLE_RAG:
499
+ # # 重新生成 RAG 向量数据库
500
+ # gen_rag_db(force_gen=True)
501
 
502
+ # # 重新加载 retriever
503
+ # RAG_RETRIEVER.pop("default")
504
+ # RAG_RETRIEVER.get(fs_id="default", config_path=WEB_CONFIGS.RAG_CONFIG_PATH, work_dir=WEB_CONFIGS.RAG_VECTOR_DB_DIR)
505
 
506
  # 更新状态
507
  status.update(label="添加科室信息成功!", state="complete", expanded=False)