demo / app.py
mujtaba025's picture
Create app.py
0142285 verified
raw
history blame contribute delete
600 Bytes
import gradio as gr
from groq import Groq
client = Groq(api_key="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")
def chat(message, history):
messages = []
for user_msg, assistant_msg in history:
messages.append({"role": "user", "content": user_msg})
messages.append({"role": "assistant", "content": assistant_msg})
messages.append({"role": "user", "content": message})
completion = client.chat.completions.create(model="llama-3.3-70b-versatile", messages=messages)
return completion.choices[0].message.content
gr.ChatInterface(fn=chat).launch()