841e14a093
- 新增 dataset_page 模块，实现数据集页面的基本布局 - 重构 main.py 中的页面加载方式，使用列表收集所有页面 - 更新主页面布局，将聊天页面作为第一个选项卡 - 调整设置页面的加载方式，直接使用函数调用
27 lines
828 B
Python
27 lines
828 B
Python
import gradio as gr
|
|
from frontend.setting_page import setting_page
|
|
from frontend import chat_page,setting_page,train_page,dataset_page
|
|
from db import initialize_db as init_db,get_engine
|
|
|
|
if __name__ == "__main__":
|
|
init_db(get_engine("workdir"))
|
|
|
|
pages = []
|
|
pages.append(setting_page())
|
|
pages.append(chat_page())
|
|
pages.append(train_page())
|
|
pages.append(dataset_page())
|
|
|
|
with gr.Blocks() as app:
|
|
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
|
|
with gr.Tabs():
|
|
with gr.TabItem("聊天"):
|
|
chat_page()
|
|
with gr.TabItem("微调"):
|
|
train_page()
|
|
with gr.TabItem("数据集"):
|
|
dataset_page()
|
|
with gr.TabItem("设置"):
|
|
setting_page()
|
|
|
|
app.launch() |