satyam001 commited on
Commit
60b00e7
·
1 Parent(s): 0bdc221

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ def main():
84
  if query:
85
  docs = VectorStore.similarity_search(query=query, k=3)
86
  llm = HuggingFaceHub(repo_id='OpenAssistant/oasst-sft-1-pythia-12b',
87
- huggingfacehub_api_token='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', model_kwargs={"temperature": 0.1, "max_new_tokens": 250})
88
  # llm = OpenAI()
89
  chain = load_qa_chain(llm=llm, chain_type="stuff")
90
  response = chain.run(input_documents=docs, question=query)
 
84
  if query:
85
  docs = VectorStore.similarity_search(query=query, k=3)
86
  llm = HuggingFaceHub(repo_id='OpenAssistant/oasst-sft-1-pythia-12b',
87
+ huggingfacehub_api_token='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', model_kwargs={"temperature": 0.1, "max_new_tokens": 500})
88
  # llm = OpenAI()
89
  chain = load_qa_chain(llm=llm, chain_type="stuff")
90
  response = chain.run(input_documents=docs, question=query)