29 lines
880 B
Python
29 lines
880 B
Python
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("## 微调")
|
|
# 获取数据集列表并设置初始值
|
|
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() |