Spaces:

j-js
/

TradingGameAI

Sleeping

j-js commited on Mar 13

Commit

9dd5333

verified ·

1 Parent(s): c27e0d2

Update retrieval_engine.py

Files changed (1) hide show

retrieval_engine.py CHANGED Viewed

@@ -6,7 +6,11 @@ import numpy as np
 class RetrievalEngine:
     def __init__(self):
-        self.dataset = load_dataset("j-js/gmat-quant-corpus", split="train")
         self.model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")

 class RetrievalEngine:
     def __init__(self):
+        self.dataset = load_dataset(
+            "json",
+            data_files="https://huggingface.co/datasets/j-js/gmat-quant-corpus/resolve/main/gmat_hf_chunks.jsonl",
+            split="train",
+        )
         self.model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")