def METHOD_NAME(work_dir='./gpt3_poetry'):
    dataset_dict = MsDataset.load('chinese-poetry-collection')
    eval_dataset = dataset_dict['test'].remap_columns({
        'text1': 'src_txt'
    }).select(range(20))
    tmp_dir = './gpt3_poetry'
    kwargs = dict(
        model=fr'{tmp_dir}/output',
        predict_datasets=eval_dataset,
        work_dir=tmp_dir,
        launcher='pytorch')
    trainer = build_trainer(default_args=kwargs)
    trainer.predict()