17行代码实现藏头诗生成器

paddlehub 让人工智能的实现非常简单。而 gradio 更是前端交互神器。两者结合，可以快速搭建你想要的人工智能 Demo 演示产品。

代码只有 17 行：

import gradio as gr
import random
import time
import paddlehub as hub

module = hub.Module(name="ernie_gen_acrostic_poetry")

with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    msg = gr.Textbox()
    clear = gr.ClearButton([msg, chatbot])

    def respond(message, chat_history):
        results = module.generate(texts=[message], use_gpu=False, beam_width=1)
        bot_message = results[0][0] #random.choice(["How are you?", "I love you", "I'm very hungry"])
        chat_history.append((message, bot_message))
        return "", chat_history

    msg.submit(respond, [msg, chatbot], [msg, chatbot])

if __name__ == "__main__":
    demo.launch()

运行后开启7860端口，用浏览器访问，效果如下：