Samarth991 commited on
Commit
1648d0f
·
verified ·
1 Parent(s): 2a5defe

Update QnA.py

Browse files
Files changed (1) hide show
  1. QnA.py +5 -3
QnA.py CHANGED
@@ -16,7 +16,7 @@ load_dotenv()
16
  def prompt_template_to_analyze_resume():
17
  template = """
18
  You are provided with the Resume of the Candidate in the context below . As an Talent Aquistion bot , your task is to provide insights about the
19
- candidate.
20
 
21
  \n\n:{context}
22
  """
@@ -65,9 +65,10 @@ def get_hugging_face_model(model_id='mistralai/Mistral-7B-Instruct-v0.2',tempera
65
  return llm
66
 
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
 
@@ -85,4 +86,5 @@ def Q_A(vectorstore,question,API_KEY):
85
 
86
  chain = create_retrieval_chain(compression_retriever, question_answer_chain)
87
  result = chain.invoke({'input':question})
88
- return result['answer']
 
 
16
  def prompt_template_to_analyze_resume():
17
  template = """
18
  You are provided with the Resume of the Candidate in the context below . As an Talent Aquistion bot , your task is to provide insights about the
19
+ candidate in precise manner.
20
 
21
  \n\n:{context}
22
  """
 
65
  return llm
66
 
67
  def Q_A(vectorstore,question,API_KEY):
68
+
69
  if API_KEY.startswith('gsk'):
70
  os.environ["GROQ_API_KEY"] = API_KEY
71
+ chat_llm = ChatGroq(model="llama3-8b-8192") # (model="gemma2-9b-it")
72
  elif API_KEY.startswith('hf'):
73
  chat_llm = get_hugging_face_model(api_key=API_KEY)
74
 
 
86
 
87
  chain = create_retrieval_chain(compression_retriever, question_answer_chain)
88
  result = chain.invoke({'input':question})
89
+ return result['answer']
90
+