Update app.py

app.py

@@ -1,6 +1,6 @@
 from datasets import load_dataset
-dataset = load_dataset("Namitg02/Test")
-print(dataset)
+data = load_dataset("Namitg02/Test",revision = "embedded"))
+print(data)
 
 from langchain.docstore.document import Document as LangchainDocument
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -10,6 +10,8 @@ docs = splitter.create_documents(str(dataset))
 from sentence_transformers import SentenceTransformer
 #from langchain_community.embeddings import HuggingFaceEmbeddings
 embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
+data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset
+
 
 from langchain_community.vectorstores import Chroma
 persist_directory = 'docs/chroma/'