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端口,用浏览器访问,效果如下: