diff --git a/server/api.py b/server/api.py index 900ba003..21c9c1fa 100644 --- a/server/api.py +++ b/server/api.py @@ -6,7 +6,7 @@ sys.path.append(os.path.dirname(os.path.dirname(__file__))) from configs.model_config import NLTK_DATA_PATH, OPEN_CROSS_DOMAIN import argparse import uvicorn -from fastapi import FastAPI +from fastapi_offline import FastAPIOffline as FastAPI from fastapi.middleware.cors import CORSMiddleware from starlette.responses import RedirectResponse from server.chat import (chat, knowledge_base_chat, openai_chat, diff --git a/webui.py b/webui.py index 07e39002..c883b100 100644 --- a/webui.py +++ b/webui.py @@ -44,7 +44,7 @@ if __name__ == "__main__": pages = {"对话": {"icon": "chat", "func": dialogue_page, }, - "知识库管理": {"icon": "database-fill-gear", + "知识库管理": {"icon": "hdd-stack", "func": knowledge_base_page, }, "模型配置": {"icon": "gear",