Manglik-R commited on
Commit
5579f09
·
1 Parent(s): 1b2a373

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,14 +30,14 @@ def pdf_changes(pdf_doc):
30
  embeddings = HuggingFaceHubEmbeddings()
31
  db = FAISS.from_documents(docs, embeddings)
32
 
33
- llm=HuggingFaceHub(repo_id="google/flan-ul2", model_kwargs={"temperature":0.1, "max_new_tokens":300})
34
  global qa
35
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever())
36
  return "Ready"
37
 
38
  def book_changes(book):
39
  db = FAISS.load_local( book , embeddings = HuggingFaceHubEmbeddings() )
40
- llm=HuggingFaceHub(repo_id="google/flan-ul2", model_kwargs={"temperature":0.1, "max_new_tokens":300})
41
  global qa
42
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever())
43
  return "Ready"
 
30
  embeddings = HuggingFaceHubEmbeddings()
31
  db = FAISS.from_documents(docs, embeddings)
32
 
33
+ llm=HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":0.1, "max_new_tokens":250})
34
  global qa
35
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever())
36
  return "Ready"
37
 
38
  def book_changes(book):
39
  db = FAISS.load_local( book , embeddings = HuggingFaceHubEmbeddings() )
40
+ llm=HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":0.1, "max_new_tokens":250})
41
  global qa
42
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever())
43
  return "Ready"