Spaces:

seawolf2357
/

rag

Runtime error

seawolf2357 commited on May 19, 2024

Commit

f4e7415

verified ·

1 Parent(s): 3cca354

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ accelerator = Accelerator()
 model = accelerator.prepare(model)
 ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
-dataset = load_dataset("lcw99/wikipedia-korean-20221001")
 data = dataset["train"]
 data = data.add_faiss_index("embeddings")

 model = accelerator.prepare(model)
 ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
+dataset = load_dataset("not-lain/wikipedia", revision="embedded")
 data = dataset["train"]
 data = data.add_faiss_index("embeddings")