Update app.py

app.py

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