Juan Martínez commited on
Commit
a42b85f
·
1 Parent(s): e44114f
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def search(text, model, ds, n):
12
 
13
  @st.cache()
14
  def get_dataset():
15
- ds = load_dataset("justinian336/salvadoran-news-embedded")
16
  ds.add_faiss_index(column="embedding")
17
  return ds
18
 
 
12
 
13
  @st.cache()
14
  def get_dataset():
15
+ ds = load_dataset("justinian336/salvadoran-news-embedded")["train"]
16
  ds.add_faiss_index(column="embedding")
17
  return ds
18