groq-chatbot / app.py
net2asif's picture
Update app.py
3ab1dbe verified
raw
history blame contribute delete
872 Bytes
import gradio as gr
from groq import Groq
import os
groq_api=os.getenv("groq_api_key")
client = Groq(api_key = groq_api)
def chat(message, history):
chat_completion = client.chat.completions.create(
#
# Required parameters
#
messages=[
{
"role": "system",
"content": "you are a helpful assistant."
},
# Set a user message for the assistant to respond to.
{
"role": "user",
"content": message,
}
],
# The language model which will generate the completion.
model="llama3-8b-8192",
temperature=0.5,
max_tokens=1024,
top_p=1,
stop=None,
stream=False,
)
return chat_completion.choices[0].message.content
demo = gr.ChatInterface(fn=chat, title="Groq-Ai-Chabot")
demo.launch(debug= True)