Chatbot / app.py
0Learn's picture
Update app.py
bc5ce27 verified
raw
history blame
629 Bytes
import os
import gradio as gr
from groq import Groq
client = Groq(
api_key=os.environ.get("GROQ_API_KEY"),
)
def chat_with_groq(user_input, additional_info=None):
chat_completion = client.chat.completions.create(
messages=[
{
"role":"user",
"content":user_input,
}
],
model="llama3-8b-8192",
)
return chat_completion.choices[0].message.content
iface = gr.ChatInterface(
fn=chat_with_groq,
title="Groq Chatbot",
description="Ask anything to the Groq powered chatbot"
)
if __name__ == "__main__":
iface.launch()