chatinterface-test-ui

Running

not-lain commited on Apr 6

Commit

5ea07e3

•

1 Parent(s): 4871902

🌘w🌖

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ TOP_K = 3
 # prepare data
 # since data is too big we will only select the first 3K lines
-data = load_dataset("not-lain/wikipedia-small-3000-embedded", subset="train")
 # index dataset
 data.add_faiss_index("embedding", device=1)

 # prepare data
 # since data is too big we will only select the first 3K lines
+data = load_dataset("not-lain/wikipedia-small-3000-embedded", split="train")
 # index dataset
 data.add_faiss_index("embedding", device=1)