Manglik-R commited on
Commit
83b1704
·
1 Parent(s): 8ef81c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -37,8 +37,7 @@ def pdf_changes(pdf_doc):
37
  return "Ready"
38
 
39
  def book_changes(book):
40
-
41
- db = torch.load(book, map_location='cpu')
42
  llm=HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":1, "max_length":1000000})
43
  global qa
44
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever(search_kwargs={"k": 3}))
 
37
  return "Ready"
38
 
39
  def book_changes(book):
40
+ db = torch.load(book, map_location=torch.device('cpu'))
 
41
  llm=HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":1, "max_length":1000000})
42
  global qa
43
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever(search_kwargs={"k": 3}))