Update app.py

app.py

@@ -13,10 +13,10 @@ import time
 token = os.environ["HF_TOKEN"]
 ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
 
-dataset = load_dataset("not-lain/wikipedia",revision = "embedded")
+dataset = load_dataset("Yoxas/statistical_literacyv2",revision = "embedded")
 
 data = dataset["train"]
-data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset
+data = data.add_faiss_index("Abstract_Embeddings") # column name that has the embeddings of the dataset
 
 
 model_id = "meta-llama/Meta-Llama-3-8B-Instruct"