Compare commits
2 Commits
513b639bce
...
841e14a093
Author | SHA1 | Date | |
---|---|---|---|
![]() |
841e14a093 | ||
![]() |
2ff077bb1c |
4
frontend/__init__.py
Normal file
4
frontend/__init__.py
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
from .chat_page import chat_page
|
||||||
|
from .setting_page import setting_page
|
||||||
|
from .train_page import train_page
|
||||||
|
from .dataset_page import dataset_page
|
9
frontend/dataset_page.py
Normal file
9
frontend/dataset_page.py
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
import gradio as gr
|
||||||
|
|
||||||
|
def dataset_page():
|
||||||
|
with gr.Blocks() as demo:
|
||||||
|
gr.Markdown("## 数据集")
|
||||||
|
with gr.Row():
|
||||||
|
with gr.Column():
|
||||||
|
pass
|
||||||
|
return demo
|
21
main.py
21
main.py
@ -1,24 +1,27 @@
|
|||||||
import gradio as gr
|
import gradio as gr
|
||||||
from frontend.setting_page import setting_page
|
from frontend.setting_page import setting_page
|
||||||
from frontend.chat_page import chat_page
|
from frontend import chat_page,setting_page,train_page,dataset_page
|
||||||
from frontend.train_page import train_page
|
|
||||||
from db import initialize_db as init_db,get_engine
|
from db import initialize_db as init_db,get_engine
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
init_db(get_engine("workdir"))
|
init_db(get_engine("workdir"))
|
||||||
|
|
||||||
setting_demo = setting_page()
|
pages = []
|
||||||
chat_demo = chat_page()
|
pages.append(setting_page())
|
||||||
train_demo = train_page()
|
pages.append(chat_page())
|
||||||
|
pages.append(train_page())
|
||||||
|
pages.append(dataset_page())
|
||||||
|
|
||||||
with gr.Blocks() as app:
|
with gr.Blocks() as app:
|
||||||
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
|
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
|
||||||
with gr.Tabs():
|
with gr.Tabs():
|
||||||
with gr.TabItem("微调"):
|
|
||||||
train_demo.render()
|
|
||||||
with gr.TabItem("聊天"):
|
with gr.TabItem("聊天"):
|
||||||
chat_demo.render()
|
chat_page()
|
||||||
|
with gr.TabItem("微调"):
|
||||||
|
train_page()
|
||||||
|
with gr.TabItem("数据集"):
|
||||||
|
dataset_page()
|
||||||
with gr.TabItem("设置"):
|
with gr.TabItem("设置"):
|
||||||
setting_demo.render()
|
setting_page()
|
||||||
|
|
||||||
app.launch()
|
app.launch()
|
Loading…
x
Reference in New Issue
Block a user