IsmaeelPandey commited on
Commit
8263d3e
·
1 Parent(s): b1717ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ PROMPT = PromptTemplate(
50
  )
51
  chain_type_kwargs = {"prompt": PROMPT}
52
 
53
- llm = HuggingFaceHub(repo_id="NousResearch/Llama-2-13b-hf", model_kwargs={"temperature":0.1, "max_new_tokens":250})
54
 
55
  qachain=RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever(), memory=memory, chain_type_kwargs=chain_type_kwargs)
56
 
 
50
  )
51
  chain_type_kwargs = {"prompt": PROMPT}
52
 
53
+ llm = HuggingFaceHub(repo_id="google/flan-ul2", model_kwargs={"temperature":0.1, "max_new_tokens":250})
54
 
55
  qachain=RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever(), memory=memory, chain_type_kwargs=chain_type_kwargs)
56