Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import pandas as pd
 from sentence_transformers import SentenceTransformer, util
+from datasets import load_dataset
+
 
 @st.cache_resource
 def load_model():
@@ -31,3 +33,13 @@ words_df = pd.DataFrame(
     columns=["word", "similarity"]
 ).sort_values(by=["similarity"], ascending=False)
 st.dataframe(words_df)
+
+
+
+@st.cache_resource
+def load_dataset():
+    dataset = load_dataset("marksverdhei/wordnet-definitions-en-2021", split="train")
+    return dataset["Word"].to_pandas()
+
+all_words = load_dataset()
+st.write(all_words)