feat(frontend): 添加模型管理页面并初始化模型相关全局变量

- 在 frontend/__init__.py 中添加 model_manage_page 模块引用
- 新增 model_manage_page.py 文件,实现模型管理页面的基本框架
- 在 global_var.py 中添加 model 和 tokenizer 全局变量
- 在 main.py 中集成模型管理页面到主应用的 Tabs 组件中
This commit is contained in:
carry 2025-04-10 17:37:45 +08:00
parent 97172f9596
commit 051d1a7535
4 changed files with 16 additions and 1 deletions

View File

@ -1,6 +1,7 @@
from .chat_page import *
from .setting_page import *
from .train_page import *
from .model_manage_page import *
from .dataset_manage_page import *
from .dataset_generate_page import *
from .prompt_manage_page import *

View File

@ -0,0 +1,9 @@
import gradio as gr
def model_manage_page():
with gr.Blocks() as demo:
gr.Markdown("## 模型管理")
with gr.Row():
with gr.Column():
pass
return demo

View File

@ -4,4 +4,7 @@ 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")
datasets = get_all_dataset("workdir")
model = None
tokenizer = None

View File

@ -10,6 +10,8 @@ if __name__ == "__main__":
with gr.Blocks() as app:
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
with gr.Tabs():
with gr.TabItem("模型管理"):
model_manage_page()
with gr.TabItem("模型推理"):
chat_page()
with gr.TabItem("模型微调"):