模型:
robinhad/gpt2-uk-conversational
该模型是在opensubtitles.org提供的电影对话数据集(未清洗)上进行训练的。
训练脚本链接: https://github.com/robinhad/ukrainian-ai 。AI演示的端到端开源链接(语音转文字转AI转语音): https://huggingface.co/spaces/robinhad/ukrainian-ai 。
from transformers import Conversation, pipeline, ConversationalPipeline conv: ConversationalPipeline = pipeline("conversational", "robinhad/gpt2-uk-conversational") text = "привіт, як тебе звати?" result = conv(Conversation(text)) # result.add_user_input() print(result)