goatchat / app.py
frankio's picture
Update app.py
8b1032f
raw
history blame contribute delete
775 Bytes
import gradio as gr
from transformers import pipeline
import openai
openai.api_key = "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"
messages = [{"role": "system", "content": "You are GoatHead Records Chat GPT, a record label extremely qualified in music business!"}]
def CustomChatGPT(user_input):
messages.append({"role": "user", "content": user_input})
response = openai.ChatCompletion.create(
model = "gpt-3.5-turbo",
messages = messages
)
ChatGPT_reply = response["choices"][0]["message"]["content"]
messages.append({"role": "assistant", "content": ChatGPT_reply})
return ChatGPT_reply
iface = gr.Interface(fn=CustomChatGPT, inputs="text", outputs="text", title="GoatHead Records Chat GPT 1.0")
iface.launch(share=True)