|
import gradio as gr |
|
from groq import Groq |
|
import os |
|
|
|
key=os.getenv("groq") |
|
|
|
client = Groq(api_key = key) |
|
|
|
def chat(message, history): |
|
|
|
|
|
chat_completion = client.chat.completions.create( |
|
|
|
|
|
|
|
messages=[ |
|
{ |
|
"role": "system", |
|
"content": "you are a helpful assistant." |
|
}, |
|
|
|
{ |
|
"role": "user", |
|
"content": message, |
|
} |
|
], |
|
|
|
|
|
model="llama3-groq-70b-8192-tool-use-preview", |
|
temperature=0.5, |
|
|
|
max_tokens=256, |
|
|
|
top_p=1, |
|
|
|
stop=None, |
|
stream=False, |
|
) |
|
|
|
return chat_completion.choices[0].message.content |
|
|
|
demo = gr.ChatInterface(fn=chat, title="Open Source chatbot") |
|
demo.launch(debug= True,share=True) |