Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def main(pdf):
|
|
43 |
|
44 |
if ask_query:
|
45 |
docs = VectorStore.similarity_search(query=ask_query, k=3)
|
46 |
-
llm = HuggingFaceHub(repo_id="
|
47 |
chain = load_qa_chain(llm=llm, chain_type='stuff')
|
48 |
response = chain.run(input_documents=docs, question=ask_query)
|
49 |
st.write(response)
|
|
|
43 |
|
44 |
if ask_query:
|
45 |
docs = VectorStore.similarity_search(query=ask_query, k=3)
|
46 |
+
llm = HuggingFaceHub(repo_id="Salesforce/xgen-7b-8k-base", model_kwargs={"temperature": 0, "max_length": 64})
|
47 |
chain = load_qa_chain(llm=llm, chain_type='stuff')
|
48 |
response = chain.run(input_documents=docs, question=ask_query)
|
49 |
st.write(response)
|