Spaces:
Sleeping
Sleeping
Commit
·
61a4f47
1
Parent(s):
19d4bc3
Added formatted dataset
Browse files- utils/data_loader.py +1 -1
utils/data_loader.py
CHANGED
|
@@ -3,7 +3,7 @@ from datasets import load_dataset
|
|
| 3 |
|
| 4 |
print("Loading dataset and indexing FAISS...") # Optional: for debugging
|
| 5 |
|
| 6 |
-
dataset = load_dataset("subashdvorak/tiktok-
|
| 7 |
data = dataset['train'].add_faiss_index('embeddings')
|
| 8 |
|
| 9 |
def load_influencer_data():
|
|
|
|
| 3 |
|
| 4 |
print("Loading dataset and indexing FAISS...") # Optional: for debugging
|
| 5 |
|
| 6 |
+
dataset = load_dataset("subashdvorak/tiktok-formatted-story", revision="embedded")
|
| 7 |
data = dataset['train'].add_faiss_index('embeddings')
|
| 8 |
|
| 9 |
def load_influencer_data():
|