ShivanshMathur007 commited on
Commit
179fb88
1 Parent(s): 90f5e23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def MOP(path):
30
  filename_to_keep = 'Dummy_standard MoP_template_new.pdf'
31
  prompt_file=delete_files_except(filename_to_keep,path)
32
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
33
- llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 2048})
34
  retriever = vector_store.as_retriever(search_type="similarity",search_kwargs={"k": len(text_chunks)})
35
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever,verbose=True)
36
  message= f"""<s> [INST] You have two documents:
 
30
  filename_to_keep = 'Dummy_standard MoP_template_new.pdf'
31
  prompt_file=delete_files_except(filename_to_keep,path)
32
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
33
+ llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 4096})
34
  retriever = vector_store.as_retriever(search_type="similarity",search_kwargs={"k": len(text_chunks)})
35
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever,verbose=True)
36
  message= f"""<s> [INST] You have two documents: