Spaces:
Runtime error
Runtime error
arborvitae
commited on
Commit
•
0078522
1
Parent(s):
1492ae5
Update app.py
Browse files
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":
|
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"
|