Update app.py

app.py

@@ -48,7 +48,7 @@ if form_button:
         st.stop()
     if not current_model:
         st.error("You didn't choose the model for the collocate selection")
-        st.stop()
+        st.stop() 
 
     if current_model == 'MODEL 1: nplus':
         model = load_w2v('models/nplus1_word2vec.bin')
@@ -63,6 +63,6 @@ if form_button:
                                           word=target_word,
                                           target_pos=target_word_pos,
                                           topn=collocate_number,
-                                          restrict_vocab=int(restrict_vocab.strip().split())
+                                          restrict_vocab=int(restrict_vocab.strip())
                                           )
     st.write(output)