Update app.py

app.py

@@ -56,7 +56,9 @@ if form_button:
     else:
         model = load_w2v('models/stihi_ru_word2vec.bin')
 
-    if not model['_'.join((target_word, target_word_pos))]:
+    try:
+        model['_'.join((target_word, target_word_pos))]:
+    except KeyError:
         st.error("The word you entered is not present in the model")
         st.stop()