使用gradio快速实现聊天机器人

我们可以使用gradio库通过低代码的方式快速实现聊天机器人界面及交互:

import gradio as gr
from ollama import chat

def predict(message,history):
	stream = chat(model="deepseek-r1:1.5b",messages=[{"role":"user","content":message}],stream=True)
	response_text = ""
	for chunk in stream:
		text = chunk["message"]["content"]
		response_text += text
		yield response_text

demo = gr.ChatInterface(fn=predict,type="messages",examples=["你是谁?",],title="聊天机器人")
demo.launch()

其界面类似如下:

更多配置可以参考官网进行配置。