ladapetrushenko commited on
Commit
7f1c71a
·
1 Parent(s): 0ae7dcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ if form_button:
56
  else:
57
  model = load_w2v('models/stihi_ru_word2vec.bin')
58
 
59
- if '_'.join((target_word, target_word_pos)) not in model.vocab:
60
  st.error("The word you entered is not present in the model")
61
  st.stop()
62
 
 
56
  else:
57
  model = load_w2v('models/stihi_ru_word2vec.bin')
58
 
59
+ if '_'.join((target_word, target_word_pos)) not in model.index_to_key:
60
  st.error("The word you entered is not present in the model")
61
  st.stop()
62