edjdhug3 commited on
Commit
389223f
1 Parent(s): bb3bf4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="google/flan-t5-xl", 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)
 
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)