Update app.py

app.py

@@ -41,14 +41,20 @@ class DocumentRetrievalAndGeneration:
 
     def create_faiss_index(self):
         all_texts = [split.page_content for split in self.all_splits]
-        embeddings = self.embeddings.encode(all_texts, convert_to_tensor=True).cpu().numpy()
-        vectordb = FAISS.from_embeddings(
-            texts=all_texts,  # Provide texts separately
-            embeddings=embeddings,  # Provide embeddings separately
-            model=self.embeddings,
-            metadatas=[{"source": f"doc_{i}"} for i in range(len(all_texts))]
+        embeddings = self.embeddings.encode(all_texts)
+        
+        # Create FAISS index
+        vector_dimension = embeddings.shape[1]
+        index = faiss.IndexFlatL2(vector_dimension)
+        index.add(embeddings)
+        
+        # Create and return FAISS object
+        return FAISS(
+            embedding_function=self.embeddings.encode,
+            index=index,
+            docstore=self.all_splits,
+            index_to_docstore_id={i: i for i in range(len(self.all_splits))}
         )
-        return vectordb
 
     def initialize_llm(self, model_id):
         quantization_config = BitsAndBytesConfig(