Update app.py

app.py

@@ -1,5 +1,5 @@
 from datasets import load_dataset
-data = load_dataset("Namitg02/Test")
+dataset = load_dataset("Namitg02/Test")
 print(data)
 
 from langchain.docstore.document import Document as LangchainDocument
@@ -10,6 +10,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 from sentence_transformers import SentenceTransformer
 #from langchain_community.embeddings import HuggingFaceEmbeddings
 embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
+data = dataset["train"]
 data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset