Update app.py

app.py

@@ -8,6 +8,7 @@ from PyPDF2 import PdfFileReader
 import validators
 import nltk
 import streamlit as st
+import pickle
 
 nltk.download('punkt')
 
@@ -153,8 +154,10 @@ def search_func(query):
     elif pdf_title:
 
         st.write(f"Document Header: {pdf_title}")
+       
 
     # Encode the query using the bi-encoder and find relevant answers
+    corpus_embeddings = pd.read_pickle("corpus_embeddings_cpu.pkl")
     question_embedding = bi_encoder.encode(query, convert_to_tensor=True)
     question_embedding = question_embedding.cpu()
     hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=2, score_function=util.dot_score)