Samarth991 commited on
Commit
e6081ef
·
verified ·
1 Parent(s): 743ff2e

Update QnA.py

Browse files
Files changed (1) hide show
  1. QnA.py +1 -1
QnA.py CHANGED
@@ -55,7 +55,7 @@ def summarize(documents,llm):
55
  return results['output_text']
56
 
57
 
58
- def get_hugging_face_model(model_id='tiiuae/Falcon3-7B-Instruct',temperature=0.01,max_tokens=2048,api_key=None):
59
  llm = HuggingFaceHub(
60
  huggingfacehub_api_token =api_key ,
61
  repo_id=model_id,
 
55
  return results['output_text']
56
 
57
 
58
+ def get_hugging_face_model(model_id='mistralai/Mixtral-8x7B-Instruct-v0.1',temperature=0.01,max_tokens=2048,api_key=None):
59
  llm = HuggingFaceHub(
60
  huggingfacehub_api_token =api_key ,
61
  repo_id=model_id,