【2022 CCF BDCI 文心大模型创意项目】DIY绘本

# 环境安装
!pip install paddlehub —upgrade
!pip install docarray
# 文心模型
!hub install ernie_zeus
!hub install ernie_vilg
import paddlehub as hub

model = hub.Module(name='ernie_zeus')

script = model.custom_generation(
    text='儿童睡前故事，有关小动物',
    min_dec_len=70,
    seq_len=1000,
    topp=0.5,
    penalty_score=1.2
)

print(script)
import paddlehub as hub    # 导入paddlehub
ernie_vilg_module = hub.Module(name='ernie_vilg')    # 导入模型

input_text = input("插图：")
results = ernie_vilg_module.generate_image(
    text_prompts=input_text,    # 提示词
    style="油画",    # 图片类型
    topk = 1,    # 图片数量，最大1
output_dir='./data/插图')    # 图片保存目录，输入插图内容后点击回车键

!pip install —upgrade paddlehub
!pip install docarray
import paddlehub as hub
from docarray import DocumentArray, Document

ernie_vilg_module = hub.Module(name='ernie_vilg')
DocumentArray([Document().load_pil_image_to_datauri(image) for image in results[:6]]).plot_image_sprites()
运行结果：导读：儿童睡前故事，小动物们的有趣故事 本文链接地址:儿童睡前故事：《可爱的小猫咪》 一天晚上，老虎妈妈在家里给孩子讲故事。她对孩子说：“宝贝啊！你看我家这些兄弟姐妹多么可爱呀！”“是啊！”小兔妹妹说道，“你瞧它们多漂亮呀！”“你真会夸奖人！”大象哥哥不服气了，“再好也没有我帅吧？”“当然啦！”小猴子弟弟插嘴道，“还是我最帅呢！因为我长得像孙悟空那样神通广大、能力超群…”听到他们争论起来，小狗和小猪都害怕极了，赶紧躲进自己房间里去了。只剩下小猫咪坐在沙发上，津津有味地听着老虎妈妈讲故事。