Update app.py

app.py

@@ -6,7 +6,6 @@ from langchain.vectorstores import FAISS
 from langchain.llms import HuggingFaceHub
 from langchain.chains import RetrievalQA
 from datasets import load_dataset
-import torch
 import os
 
 key = os.environ.get('RLS')
@@ -37,7 +36,7 @@ def pdf_changes(pdf_doc):
     return "Ready"
 
 def book_changes(book):
-    db = torch.load(book, map_location=torch.device('cpu'))
+    db = FAISS.load_local( book , embeddings = HuggingFaceHubEmbeddings() )
     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}))
@@ -77,7 +76,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             pdf_doc = gr.File(label="Load a PDF", file_types=['.pdf'], type="file")
             load_pdf = gr.Button("Load PDF")
-            Books = gr.Dropdown(label="Books", choices=[("Book 1","Book1.pkl") , ("Book 2","Book2.pkl"),("Book 2","Book2.pkl")] )
+            Books = gr.Dropdown(label="Books", choices=[("Book 1","Book1.faiss") , ("Book 2","Book2.faiss") , ("Book 3","Book3.faiss")] )
             langchain_status = gr.Textbox(label="Status", placeholder="", interactive=False)
         chatbot = gr.Chatbot([], elem_id="chatbot").style(height=350)
         question = gr.Textbox(label="Question", placeholder="Type your question and hit Enter ")