arborvitae commited on
Commit
0078522
1 Parent(s): 1492ae5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def pdf_changes(pdf_doc, repo_id):
26
  embeddings = HuggingFaceHubEmbeddings()
27
  db = Chroma.from_documents(texts, embeddings)
28
  retriever = db.as_retriever()
29
- llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature":0.1, "max_new_tokens":250})
30
  global qa
31
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True)
32
  return "Ready"
 
26
  embeddings = HuggingFaceHubEmbeddings()
27
  db = Chroma.from_documents(texts, embeddings)
28
  retriever = db.as_retriever()
29
+ llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature":0.1, "max_new_tokens":1048})
30
  global qa
31
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True)
32
  return "Ready"