Update app.py

app.py

@@ -25,15 +25,6 @@ dataset = load_dataset("Namitg02/Test", split='train', streaming=False)
 #Returns a list of dictionaries, each representing a row in the dataset.
 print(dataset[1])
 length = len(dataset)
-df = pd.DataFrame(dataset)
-
-embeddings = embedding_model.encode(dataset["text"])
-print(embeddings)
-
-df['embeddings'] = embeddings
-dataset = Dataset.from_pandas(df)
-print(dataset[1])
-
 
 #Itemdetails = dataset.items()
 #print(Itemdetails)
@@ -42,6 +33,16 @@ embedding_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
 #embedding_model = HuggingFaceEmbeddings(model_name = "mixedbread-ai/mxbai-embed-large-v1")
 #all-MiniLM-L6-v2, BAAI/bge-base-en-v1.5,infgrad/stella-base-en-v2, BAAI/bge-large-en-v1.5 working with default dimensions
 
+df = pd.DataFrame(dataset)
+display(df)
+embeddings = embedding_model.encode(dataset["text"])
+print(embeddings)
+df['embeddings'] = embeddings
+display(df)
+dataset = Dataset.from_pandas(df)
+print(dataset[1])
+print(dataset[2])
+
 #doc_func = lambda x: x.text
 #dataset = list(map(doc_func, dataset))