051d1a7535
- 在 frontend/__init__.py 中添加 model_manage_page 模块引用 - 新增 model_manage_page.py 文件，实现模型管理页面的基本框架 - 在 global_var.py 中添加 model 和 tokenizer 全局变量 - 在 main.py 中集成模型管理页面到主应用的 Tabs 组件中
10 lines
300 B
Python
10 lines
300 B
Python
from db import get_sqlite_engine, get_prompt_tinydb, get_all_dataset
|
|
from tools import scan_docs_directory
|
|
|
|
prompt_store = get_prompt_tinydb("workdir")
|
|
sql_engine = get_sqlite_engine("workdir")
|
|
docs = scan_docs_directory("workdir")
|
|
datasets = get_all_dataset("workdir")
|
|
|
|
model = None
|
|
tokenizer = None |