Spaces:
Sleeping
Sleeping
Update QnA.py
Browse files
QnA.py
CHANGED
@@ -67,7 +67,7 @@ def get_hugging_face_model(model_id='mistralai/Mistral-7B-Instruct-v0.2',tempera
|
|
67 |
def Q_A(vectorstore,question,API_KEY):
|
68 |
if API_KEY.startswith('gsk'):
|
69 |
os.environ["GROQ_API_KEY"] = API_KEY
|
70 |
-
chat_llm = ChatGroq(model="llama3-8b-8192")
|
71 |
elif API_KEY.startswith('hf'):
|
72 |
chat_llm = get_hugging_face_model(api_key=API_KEY)
|
73 |
|
|
|
67 |
def Q_A(vectorstore,question,API_KEY):
|
68 |
if API_KEY.startswith('gsk'):
|
69 |
os.environ["GROQ_API_KEY"] = API_KEY
|
70 |
+
chat_llm = ChatGroq(model="gemma2-9b-it") #(model="llama3-8b-8192")
|
71 |
elif API_KEY.startswith('hf'):
|
72 |
chat_llm = get_hugging_face_model(api_key=API_KEY)
|
73 |
|