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

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

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

代码只有 17 行：

import gradio as gr
import random
import time
import paddlehub as hubmodule = 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 = ate(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_historymsg.submit(respond, [msg, chatbot], [msg, chatbot])if __name__ == "__main__":demo.launch()

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