diff --git a/frontend/train_page.py b/frontend/train_page.py index 44e2d82..62b5d8f 100644 --- a/frontend/train_page.py +++ b/frontend/train_page.py @@ -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 - return demo \ No newline at end of file + # 获取数据集列表并设置初始值 + 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() \ No newline at end of file