Spaces:
Sleeping
Sleeping
import os | |
import gradio as gr | |
from groq import Groq | |
client = Groq( | |
# This is the default and can be omitted | |
api_key=os.environ.get("GROQ_API_KEY"), | |
) | |
def chat_with_groq(user_input, additional_info=None): # Added additional_info parameter | |
chat_completion = client.chat.completions.create( | |
messages=[ | |
{ | |
"role": "system", | |
"content": "you are a helpful assistant." | |
}, | |
{ | |
"role": "user", | |
"content": user_input, | |
} | |
], | |
model="mixtral-8x7b-32768", | |
) | |
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() |