Update app.py

app.py

@@ -49,12 +49,12 @@ def main():
         for i, (class_, probability) in enumerate(zip(top_classes, top_probabilities)):
             suggested_word = data[data.Label == class_].Word.values[0]
             button_text = f"Did you mean '{suggested_word}'?"
-        if st.button(button_text):
-            definition = dataa[dataa['Word'].str.lower() == suggested_word.lower()]['Definition'].values
-            if len(definition) > 0:
-                st.write(f"The dictionary meaning of '{suggested_word}' is: {definition[0]}")
-            else:
-                st.write(f"No definition found for '{suggested_word}' in the dictionary.")
+            if st.button(button_text):
+                definition = dataa[dataa['Word'] == suggested_word]['Definition'].values
+                if len(definition) > 0:
+                    st.write(f"The dictionary meaning of '{suggested_word}' is: {definition[0]}")
+                else:
+                    st.write(f"No definition found for '{suggested_word}' in the dictionary.")
 
 if __name__ == "__main__":
     main()