Update app.py

app.py

@@ -5,6 +5,7 @@ import faiss
 from PyPDF2 import PdfReader
 from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer, AutoModel
+import torch  # Import torch for tensor operations
 from langchain.vectorstores import FAISS
 from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.chains import RetrievalQA
@@ -71,7 +72,7 @@ def create_vector_store(text):
 
     def embed(sentence):
         tokens = tokenizer(sentence, return_tensors="pt", truncation=True, padding=True)
-        with torch.no_grad():
+        with torch.no_grad():  # Use torch for no_grad context
             embeddings = model(**tokens).last_hidden_state.mean(dim=1).numpy()
         return embeddings