我正在参与「兔了个兔」创意投稿大赛,详情请看:「兔了个兔」创意投稿大赛

前言

这不兔年到了,薅薅兔毛;

小兔子现在很烦恼,因为今天小兔子的教师给他们布置了课后作业,要作一首包括 “兔” 字的诗,但是小兔子现在毫无思路,让我们来帮帮小兔子;

前不久在网络爆火的 ChatGPT,被网友们玩出了各种把戏,基于此练习了一个能作中文古诗词的 GPT 预练习模型,在这儿作首诗,帮小兔子完成作业;

正文

先导入 tokenizer 对数据进行预处理:

poem_tokenizer = BertTokenizer.from_pretrained(checkpoint)

然后编码数据,并转换为 PyTorch Tensor:

index_token = poem_tokenizer.encode(text)
tokens_tensor = torch.tensor([index_token])

【AI】用 GPT 帮小兔子写首诗

然后加载模型:

poem_model = GPT2LMHeadModel.from_pretrained(checkpoint)

预测一切的输出:

poem_model.eval()
with torch.no_grad():
    outputs = poem_model(tokens_tensor)

【AI】用 GPT 帮小兔子写首诗

最后运用 argmax 以及反向解码取得终究成果:

【AI】用 GPT 帮小兔子写首诗

不过仅仅回来文字成果的话,仍是比较的单一,这儿可以跟前端结合起来,效果图如下:

【AI】用 GPT 帮小兔子写首诗

代码如下:

<link rel="preconnect" href="https://fonts.googleapis.com">
<link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
<link href="https://fonts.googleapis.com/css2?family=Nosifer&display=swap" rel="stylesheet">
<link href="https://fonts.googleapis.com/css2?family=Liu+Jian+Mao+Cao&family=Padyakke+Expanded+One&display=swap" rel="stylesheet">
<div style="background:url('https://p6-juejin.byteimg.com/tos-cn-i-k3u1fbpfcp/082a37b3aa4f43ffbb318b0994b5ea2f~tplv-k3u1fbpfcp-watermark.image?') no-repeat; width:800px; height:600px">
  <div class="font3" style="padding-left:25%">
    <p style="padding-top:40px">
      <i>
        兔蟾仙去不留迹,<br>
        万古长庚无尽明。<br>
        自昔不多司马相,<br>
        后来谁识孔方兄。<br>
      </i>
      <h1 style="padding-left:160px">sidiot</h1>
    </p>
  </div>
</div>

字体来自于 Google Fonts,怎么操作的话可以参考这个教程

以下是在码上的运转效果,大家可以自行试试:

【AI】用 GPT 帮小兔子写首诗

TIP

不太会前端,随便搞搞,不要介意;

后记

以上就是【AI】用 GPT 帮小兔子写首诗的全部内容了。

期望本篇博文对大家有所协助!

系列文章:

  • 【AI】用 GPT 帮小兔子作首诗
  • 【AI】用 GPT 帮小兔子写文章
  • 【AI】用 GPT 帮小兔子画肖像

上篇精讲:【NLP】入门(五):TF-IDF(拓展篇)

我是,等待你的重视;

创造不易,请多多支撑;

系列专栏:AI NLP