Update app.py
Browse files
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 |
-
|
500 |
-
|
501 |
|
502 |
-
|
503 |
-
|
504 |
-
|
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)
|