Update app.py

app.py

@@ -5,17 +5,18 @@ from datasets import load_dataset
 dataset = load_dataset("marksverdhei/wordnet-definitions-en-2021")
 
 # Streamlit app
-st.title("TouchDown Buddy Chatbot")
+st.title("Touchdown Buddy Chatbot")
 
 # Input question
 question = st.text_input("Enter your question:")
 
 # Search for the corresponding definition
 if question:
-    definitions = dataset["train"].filter(lambda example: question.lower() in example["gloss"].lower())
-    if len(definitions) > 0:
-        st.write("Definitions:")
-        for i, definition in enumerate(definitions):
-            st.write(f"{i+1}. {definition['gloss']}")
+    for example in dataset["train"]:
+        if "definition" in example:
+            definitions = example["definition"]
+            if question.lower() in definitions.lower():
+                st.write("Definition:", definitions)
+                break
     else:
         st.write("Sorry, the answer to your question is not available.")