Spaces:

seawolf2357
/

rag

Runtime error

seawolf2357 commited on May 1, 2024

Commit

03d808d

verified ·

1 Parent(s): f544f96

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("jojo0217/korean_safe_conversation", revision="embedded")
 data = dataset["train"]
 data = data.add_faiss_index("embeddings")

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