519a5f3773
- 新增 chat_page.py、setting_page.py 和 train_page.py 文件，分别实现聊天、设置和微调页面的基本布局 - 添加 main.py 文件，集成所有页面并创建主应用 - 在 requirements.txt 中添加 gradio 依赖
23 lines
678 B
Python
23 lines
678 B
Python
import gradio as gr
|
|
from frontend.setting_page import setting_page
|
|
from frontend.chat_page import chat_page
|
|
from frontend.train_page import train_page
|
|
def main():
|
|
setting_demo = setting_page()
|
|
chat_demo = chat_page()
|
|
train_demo = train_page()
|
|
|
|
with gr.Blocks() as app:
|
|
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
|
|
with gr.Tabs():
|
|
with gr.TabItem("微调"):
|
|
train_demo.render()
|
|
with gr.TabItem("聊天"):
|
|
chat_demo.render()
|
|
with gr.TabItem("设置"):
|
|
setting_demo.render()
|
|
|
|
app.launch()
|
|
|
|
if __name__ == "__main__":
|
|
main() |