当前位置:   article > 正文

AI系列:用Python streamlit实现简单的AI 大模型对话对比器_streamlit 大模型

streamlit 大模型

前言

如何实现与AI大模型的对话?一种选择是登陆AI大模型厂商提供的对话网站或者App。另外,目前网络上还有很多开源的实现,比如ChatboxChathubChatALL等等。这些项目大多使用Typescript, Javascript,提供配置功能,大模型人设等功能。
这篇文章里,我们使用Python的几十行代码来实现一个简单的AI 大模型对话对比器

功能介绍

这个AI大模型对话对比器包含与大模型对话和多个大模型对比的功能。

  • 提供浏览器界面,供用户输入提示词和显示对话生成结果。
  • 可以对比多个大模型 (目前代码示例为GPT3和GPT4)。
  • 保存对话上下文直至手动清空。

界面如下:
AI大模型对话&对比器
大模型本身没有记忆功能。从截图中可以看到,程序本身可以记忆上下文,上文中的英文被输入给大模型,并在用户的要求下被大模型翻译成下文中的中文。

代码实现

这里要介绍一下Streamlit,使用它根本不需要任何前端代码,只需要纯Python就可以快速构建漂亮的Web应用。Streamlit很流行,甚至AI大模型也支持Streamlit代码。

文中这个简单的系统-AI 大模型对话对比器,只使用了几十行的Python代码,按照前端功能和后端功能分成两个文件。
你可以通过访问我的Github直接下载代码。

前端功能

前端功能主要包括输入提示词和显示大模型对话生成内容。在用户清空对话之前,保存对话的上下文,并作为输入调用后端功能。

首先确保在调试运行之前安装了streamlit:

$ pip install streamlit
  • 1

前端功能的python文件 - aitalk.py

__author__ = 'liyane'

import streamlit as st
from aibackend import models, get_chat_response_from_all_models

# define page / 定义页面
st.set_page_config(layout="wide")
st.title("
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/771367
推荐阅读
相关标签