Update app.py

app.py

@@ -59,10 +59,15 @@ if form_button:
     else:
         model = load_w2v('models/stihi_ru_word2vec.bin')
 
+    try:
+        restrict_vocab = int(restrict_vocab.strip())
+    except ValueError:
+        restrict_vocab = None
+
     output = get_collocates_for_word_type(model=model,
                                           word=target_word,
                                           target_pos=target_word_pos,
                                           topn=collocate_number,
                                           restrict_vocab=restrict_vocab)
-                                          )
+                                          
     st.write(output)