Исправление кандидатов.

app.py

@@ -75,6 +75,10 @@ text = preprocess([title + ". " + abstract])[0]
 if not text is None and len(text) > 0:
     #keywords = get_candidates(text, main_nlp)
     keywords = get_keywords(text, main_nlp, main_model, main_tokenizer)
+    if keywords is None:
+        st.markdown("Вы ввели либо слишком короткий, либо грамматически некорректный текст. Попробуйте еще раз.")
+        break
+
     labels = [kw[0].replace(' ', '\n') for kw in keywords]
     scores = [kw[1] for kw in keywords]
     #st.markdown(f"{keywords}")
@@ -97,4 +101,4 @@ if not text is None and len(text) > 0:
 
     st.pyplot(fig)
 else:
-    st.markdown("Please, try to enter something.")
+    st.markdown("Пожалуйста, введите что-нибудь.")

utils/utils.py

@@ -121,8 +121,11 @@ def score_candidates(text, candidates, model, tokenizer):
 
 
 def get_keywords(text, nlp, model, tokenizer, top=0.95, max_words=None):
-    candidates = get_candidates(text, nlp)
-    score = score_candidates(text, candidates, model, tokenizer)
+    try:
+        candidates = get_candidates(text, nlp)
+        score = score_candidates(text, candidates, model, tokenizer)
+    except Exception as ex:
+        return None
     
     candidates_scored = [(candidates[index], score[index]) for index in score.argsort()[::-1]]