Spaces:

abokbot
/

wikipedia-search-engine

Running

abokbot commited on Jun 4, 2023

Commit

988f448

•

1 Parent(s): fdde008

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ st_model_load.text("")
 @st.cache_resource
 def load_wikipedia_dataset():
     print("Loading wikipedia dataset...")
-    dataset = load_dataset("abokbot/wikipedia-first-paragraph")
     return dataset
 dataset = load_wikipedia_dataset()

 @st.cache_resource
 def load_wikipedia_dataset():
     print("Loading wikipedia dataset...")
+    dataset = load_dataset("abokbot/wikipedia-first-paragraph")["train"]
     return dataset
 dataset = load_wikipedia_dataset()