模型:

robinhad/gpt2-uk-conversational

英文

乌克兰AI聊天机器人alpha版本发布

该模型是在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)