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)
|