feat(frontend): 添加数据集选择功能到训练页面
- 在 train_page.py 中添加数据集选择下拉框 - 从全局变量中获取数据集列表并设置初始值 - 添加交互性和自定义值支持
This commit is contained in:
parent
8a9a080745
commit
611904cef9
@ -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()
|
Loading…
x
Reference in New Issue
Block a user