subashdvorak
commited on
Update shared_resources.py
Browse files- shared_resources.py +1 -1
shared_resources.py
CHANGED
@@ -16,7 +16,7 @@ class SharedResources:
|
|
16 |
self.sentence_transformer = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
|
17 |
|
18 |
# Load the dataset
|
19 |
-
self.dataset = load_dataset("subashdvorak/tiktok-story-
|
20 |
self.data = self.dataset["train"]
|
21 |
self.data = self.data.add_faiss_index("embeddings")
|
22 |
|
|
|
16 |
self.sentence_transformer = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
|
17 |
|
18 |
# Load the dataset
|
19 |
+
self.dataset = load_dataset("subashdvorak/tiktok-story-data3", revision="embedded")
|
20 |
self.data = self.dataset["train"]
|
21 |
self.data = self.data.add_faiss_index("embeddings")
|
22 |
|