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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def main():
83
 
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")
 
83
 
84
  if query:
85
  docs = VectorStore.similarity_search(query=query, k=3)
86
+ llm = HuggingFaceHub(repo_id='google/flan-ul2',
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")