Update app.py

app.py

@@ -13,9 +13,9 @@ import time
 token = os.environ["HF_TOKEN"]
 ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
 
-dataset = load_dataset("not-lain/wikipedia", "Yoxas/statistical_literacy",revision = "embedded")
+dataset = load_dataset("not-lain/wikipedia",revision = "embedded")
 
-data = dataset["train", "10kstats"]
+data = dataset["train"]
 data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset