f655936741
- 新增 init_global_var 函数，用于统一初始化所有全局变量 - 修改 get_prompt_store、get_sql_engine、get_docs 和 get_datasets 函数，使用新的全局变量初始化逻辑 - 更新 main.py 中的代码，使用新的 init_global_var 函数替代原有的单独初始化方法
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
import gradio as gr
|
|
from frontend.setting_page import setting_page
|
|
from frontend import *
|
|
from db import initialize_sqlite_db, initialize_prompt_store
|
|
from global_var import init_global_var, get_sql_engine, get_prompt_store
|
|
|
|
if __name__ == "__main__":
|
|
init_global_var()
|
|
initialize_sqlite_db(get_sql_engine())
|
|
initialize_prompt_store(get_prompt_store())
|
|
with gr.Blocks() as app:
|
|
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
|
|
with gr.Tabs():
|
|
with gr.TabItem("模型管理"):
|
|
model_manage_page()
|
|
with gr.TabItem("模型推理"):
|
|
chat_page()
|
|
with gr.TabItem("模型微调"):
|
|
train_page()
|
|
with gr.TabItem("数据集生成"):
|
|
dataset_generate_page()
|
|
with gr.TabItem("数据集管理"):
|
|
dataset_manage_page()
|
|
with gr.TabItem("提示词模板管理"):
|
|
prompt_manage_page()
|
|
with gr.TabItem("设置"):
|
|
setting_page()
|
|
|
|
app.launch() |