70b64dc3d3
- 将 initialize_db 函数重命名为 initialize_sqlite_db，以明确该函数专用于 SQLite 数据库 - 更新相关模块和文件中的引用，以确保代码一致性 - 此修改旨在提高代码的可读性和维护性，特别是未来可能接入多种数据库时
22 lines
707 B
Python
22 lines
707 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_sqlite_db
|
|
from global_var import sql_engine
|
|
|
|
if __name__ == "__main__":
|
|
initialize_sqlite_db(sql_engine)
|
|
|
|
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() |