• 相声捧哏生成器

相声捧哏生成器

基于PaddleNLP完成的相声捧哏生成器。
相声中，捧哏的输出内容可以视为单轮对话生成任务。
通过收集相声数据集，微调UNIMO模型，最终模型可以输入逗哏，输出对应的捧哏。

AISTUDIO体验地址：https://aistudio.baidu.com/aistudio/projectdetail/3502884

项目的实现效果如下:

逗哏输入：吃个糖糕粘牙

环境要求：

paddlepaddle >= 2.2.2
paddlenlp >= 2.5.0
Python >=3.7

📖 使用说明

• 由于涉及input输入，建议在控制台运行

1. 安装paddlenlp及paddlepaddle环境
2. 下载数据并解压至 ./data_crosstalk 目录下
3. 运行 main.py 完成模型训练，模型保存为 ./unimo_result/model.pdparams
4. 运行 infer.py 进行体验

参考资料：

• https://zhuanlan.zhihu.com/p/42568781
• https://aistudio.baidu.com/aistudio/projectdetail/2341543