脱单“神助攻” | 情话生成器,了解 python 就能玩
允中 发自 凹非寺
量子位 编辑 | 公众号 QbitAI



AI情话实测效果展示
我也说不出来,
我心里明白,
我宁愿跟你吵架也不愿意去爱别人。



AI藏头诗实测效果展示
爱闲能有几人闲。
秋来莫作功名看,
香满南山雪满山。


https://www.paddlepaddle.org.cn/hub/scene/aiwriting
这是怎么做到的?


想知道怎么用吗?
import paddlehub as hub
#导入预训练模型
module = hub.Module(name="ernie_gen_lover_words")
#准备输入开头数据
test_texts = ['情人节']
#执行文本生成
results = module.generate(texts=test_texts, use_gpu=True, beam_width=5)
#打印输出结果
for result in results:
print(result)
[2020-08-21 20:58:08,095] [ INFO] - Installing ernie_gen_lover_words module
[2020-08-21 20:58:08,097] [ INFO] - Module ernie_gen_lover_words already installed in /home/aistudio/.paddlehub/modules/ernie_gen_lover_words
['情人节,我愿做一条鱼,任你红烧、白煮、清蒸,然后躺在你温柔的胃里。', '情人节,对你的思念太重,压断了电话线,烧坏了手机卡,掏尽了钱包袋,吃光了安眠药,哎!可我还是思念你。', '情人节,对你的思念太重,压断了电话线,烧坏了手机卡,掏尽了钱包袋,吃光了安眠药,哎!可我还是思念你,祝你情人节快乐!', '情人节,对你的思念太重,压断了电话线,烧坏了手机卡,掏尽了钱包袋,吃光了安眠药,唉!可我还是思念你,祝你情人节快乐!', '情人节,对你的思念太重,压断了电话线,烧坏了手机卡,掏尽了钱包袋,吃光了安眠药,哎!可是我还是思念你。']
!hub serving start -m ernie_gen_lover_words -p 8866
import requests
import json
# 发送HTTP请求
data = {'texts':['情人节'],
'use_gpu':False, 'beam_width':5}
headers = {"Content-type": "application/json"}
url = "http://127.0.0.1:8866/predict/ernie_gen_lover_words"
r = requests.post(url=url, headers=headers, data=json.dumps(data))
# 保存结果
results = r.json()["results"]
for result in results:
print(result)

更多预训练模型,马上了解PaddleHub

-- 1、在线代码编辑器,可以分享给任何人 -- 2、Python 造假数据,用Faker就够了 -- 3、在Python中玩转Json数据
-- 留下你的“在看”呗!
评论