Samarth991 commited on
Commit
8fc84f6
·
verified ·
1 Parent(s): 85a522e

Update QnA.py

Browse files
Files changed (1) hide show
  1. QnA.py +1 -1
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