Update app.py

app.py

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