gradio_chatbot / responser.py
rumaisa1054's picture
Create responser.py
fa4d811 verified
raw
history blame contribute delete
478 Bytes
import os
from groq import Groq
def res(prompt):
prompt = "Being a reknowned Doctor, Please Answer the question : " + prompt
client = Groq(
api_key=os.environ["key"],
)
chat_completion = client.chat.completions.create(
messages=[
{
"role": "user",
"content": prompt,
}
],
model="llama3-8b-8192",
)
return (chat_completion.choices[0].message.content)