AI知识库

53AI知识库

学习大模型的前沿技术与行业应用场景


【干货】教你搭建 Ollama + Gradio 聊天机器人
发布日期:2024-10-30 21:41:56 浏览次数: 1721 来源:码农随心笔记


本篇将为各位同学介绍Gradio,它是一个搭建可视化界面的Python库。Gradio是一个强大且易用的工具,可以帮助开发者快速构建交互式的机器学习模型展示界面。它提供了一个简单易用的界面搭建框架,让用户可以很容易地将模型部署为一个交互式的Web应用程序,而无需编写任何前端代码。
Gradio支持各种类型的输入数据,包括文本、图像、音频和视频,让用户可以通过简单的拖放操作来测试和调整模型。用户可以自定义界面的样式和布局,添加各种控件来与模型交互,比如滑块、文本框、下拉菜单等。
通过Gradio,用户可以快速构建一个漂亮的Web应用程序,用于展示自己的机器学习模型,与其他人分享模型的功能和性能,或者用于实际的应用场景中。Gradio还支持部署本地Web应用、Jupyter notebook插件或者在线部署到Gradio Hub上。
想深入了解的同学可以参考Gradio 的官网文档 : https://www.gradio.app/guides/quickstart

Gradio github:https://github.com/gradio-app/gradio


01
安装ollama + llama3.1:8b


本次我们将搭建本地聊天机器人,需要安装Ollama和llama3.1,有兴趣的同学,也可以选用其它的大模型。这里安装比较简单,根据官网指引,输入以下指令:
1. curl -fsSL https://ollama.com/install.sh | sh2. ollama pull llama3.1:8b

如果想了解详细的安装过程,请看此篇:

【干货】手把手教你搭建Ollama+OpenWebUI


02
搭建 Gradio 环境


Gradio是一个 Python 库,我们用miniconda搭建Python虚拟环境进行安装。以下指令,在终端中逐条输入:
source ~/miniconda3/bin/activateconda create -n charbot_1 python=3.12


完成后继续以下指令,在虚拟环境中安装Gradio 。

conda activate charbot_1pip install gradio



03
代码


完成以上安装后,我们可以新建charbot_1.py文件, 把以下代码放入此文件中,并且保存home/charbot_1 文件夹中。

import requestsimport jsonimport gradio as gr

url = "http://localhost:11434/api/generate"
headers = {'Content-Type': 'application/json'}

conversation_history = []
def generate_response(prompt):conversation_history.append(prompt)
full_prompt = "\n".join(conversation_history)
data = {"model": "llama3.1:8b", "stream": False,"prompt": full_prompt}

response =requests.post(url, headers=headers, data=json.dumps(data))

if response.status_code == 200:response_txt = response.textdata = json.loads(response_txt)actual_response = data["response"]conversation_history.append(actual_response)return actual_responseelse:print("Error:", response.status_code, response.text)

iface = gr.Interface(fn=generate_response,inputs=["text"],outputs=["text"])

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


04
运行


1. 先运行 ollama 把模型启动,在终端中运行

ollama run llama3.1:8B


2. 另开一个终端,进入python虚拟环境运行 charbot_1

python3 charbot_1


3. 在浏览器中打开 

地址栏输入127.0.0.1:7860,即可看到Gradio页面了,我们还可以在输入栏输入问题,向大模型提问。

本例只是简单地介绍了Gradio这个工具的聊天机器人用例,界面也比较简单,其实Gradio还可以搭建更复杂,更华丽的多模态大模型对话界面,包括通过对话生成图像,动画等等,后续有机会将带大家深入地了解Gradio,让它为我们实现更多的功能。


  --THE END--  


53AI,企业落地应用大模型首选服务商

产品:大模型应用平台+智能体定制开发+落地咨询服务

承诺:先做场景POC验证,看到效果再签署服务协议。零风险落地应用大模型,已交付160+中大型企业

联系我们

售前咨询
186 6662 7370
预约演示
185 8882 0121

微信扫码

与创始人交个朋友

回到顶部

 
扫码咨询