feat(frontend): 添加数据集选择功能到训练页面

- 在 train_page.py 中添加数据集选择下拉框
- 从全局变量中获取数据集列表并设置初始值
- 添加交互性和自定义值支持
This commit is contained in:
carry 2025-04-11 19:43:34 +08:00
parent 8a9a080745
commit 611904cef9

View File

@ -1,9 +1,29 @@
import gradio as gr
import sys
from pathlib import Path
sys.path.append(str(Path(__file__).resolve().parent.parent))
from global_var import get_model, get_tokenizer, get_datasets
def train_page():
with gr.Blocks() as demo:
gr.Markdown("## 微调")
with gr.Row():
with gr.Column():
pass
# 获取数据集列表并设置初始值
datasets_list = [str(ds["name"]) for ds in get_datasets().all()]
initial_dataset = datasets_list[0] if datasets_list else None
dataset_dropdown = gr.Dropdown(
choices=datasets_list,
value=initial_dataset, # 设置初始选中项
label="选择数据集",
allow_custom_value=True,
interactive=True
)
return demo
if __name__ == "__main__":
from global_var import init_global_var
init_global_var("workdir")
demo = train_page()
demo.queue()
demo.launch()