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:
parent
97172f9596
commit
051d1a7535
@ -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 *
|
9
frontend/model_manage_page.py
Normal file
9
frontend/model_manage_page.py
Normal 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
|
@ -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
|
Loading…
x
Reference in New Issue
Block a user