Update app.py

app.py

@@ -37,9 +37,9 @@ st.dataframe(words_df)
 
 
 @st.cache_resource
-def load_dataset():
-    dataset = load_dataset("marksverdhei/wordnet-definitions-en-2021")
+def load_words_dataset():
+    dataset = load_dataset("marksverdhei/wordnet-definitions-en-2021", split="train")
     return dataset["Word"].to_pandas()
 
-all_words = load_dataset()
+all_words = load_words_dataset()
 st.write(all_words)