ChatSherman2.0 / app.py
ShermanAI's picture
Update app.py
792173c
raw
history blame
1.56 kB
import subprocess
subprocess.check_call(["pip", "install", "-q", "gradio", "transformers", "python-dotenv"])
subprocess.check_call(["pip", "install", "-q", "openai"])
import gradio as gr
from transformers import TFAutoModelForCausalLM, AutoTokenizer
import openai
from dotenv import load_dotenv
import os
# load environment variables from .env file
load_dotenv()
# access the value of the OPENAI_API_KEY environment variable
api_key = os.getenv("OPENAI_API_KEY")
# install required packages
subprocess.check_call(["pip", "install", "-q", "gradio", "transformers", "python-dotenv", "openai"])
def openai_chat(prompt):
completions = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=1024,
n=1,
temperature=0.5,
)
message = completions.choices[0].text
return message.strip()
def chatbot(input, history=[]):
output = openai_chat(input)
history.append((input, output))
return output, history
# define Gradio interface
inputs = [
gr.inputs.Textbox(label="Input"),
gr.outputs.Label(type="hidden", label="Chat history", default=[])
]
outputs = [
gr.outputs.Textbox(label="Output"),
gr.outputs.Label(type="hidden", label="Updated chat history")
]
iface = gr.Interface(fn=chatbot,
inputs=inputs,
outputs=outputs,
title="OpenAI Chatbot",
description="A chatbot powered by OpenAI's GPT-3",
theme="compact")
# launch the interface
iface.launch()