diff --git a/frontend/__init__.py b/frontend/__init__.py index 7480f89..a7b2d6a 100644 --- a/frontend/__init__.py +++ b/frontend/__init__.py @@ -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 * \ No newline at end of file diff --git a/frontend/model_manage_page.py b/frontend/model_manage_page.py new file mode 100644 index 0000000..5126aad --- /dev/null +++ b/frontend/model_manage_page.py @@ -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 \ No newline at end of file diff --git a/global_var.py b/global_var.py index b35ab07..328cb13 100644 --- a/global_var.py +++ b/global_var.py @@ -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") \ No newline at end of file +datasets = get_all_dataset("workdir") + +model = None +tokenizer = None \ No newline at end of file diff --git a/main.py b/main.py index c3437e5..3797c1e 100644 --- a/main.py +++ b/main.py @@ -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("模型微调"):