Spaces:
Build error
Build error
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-data2", revision="embedded")
|
| 20 |
self.data = self.dataset["train"]
|
| 21 |
self.data = self.data.add_faiss_index("embeddings")
|
| 22 |
|