Update app.py

app.py

@@ -6,8 +6,7 @@ from langchain.vectorstores import FAISS
 from langchain.llms import HuggingFaceHub
 from langchain.chains import RetrievalQA
 from datasets import load_dataset
-import pickle
-import pickletools
+import pytorch
 import os
 
 key = os.environ.get('RLS')
@@ -39,8 +38,7 @@ def pdf_changes(pdf_doc):
 
 def book_changes(book):
 
-    with open( book, 'rb') as f:
-        data = pickle.load(f)
+    db = torch.load(book, map_location='cpu')
     llm=HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":1, "max_length":1000000})
     global qa 
     qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff",  retriever=db.as_retriever(search_kwargs={"k": 3}))