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

app.py

@@ -75,30 +75,30 @@ 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}")
-
-    # Топ 5 слов.
-    top = 5
-    top = min(len(labels), top)
-    st.markdown("Топ %d ключевых слов: **%s**" % (top, ', '.join(labels[0:5])))
-
-    # График важности слов.
-    fig, ax = plt.subplots(figsize=(8, len(labels)))
-    ax.set_title("95% самых важных ключевых слов")
-    ax.grid(color='#000000', alpha=0.15, linestyle='-', linewidth=1, which='major')
-    ax.grid(color='#000000', alpha=0.1, linestyle='-', linewidth=0.5, which='minor')
-
-    bar_width = 0.75
-    indexes = -np.arange(len(labels))
-    ax.barh(indexes, scores, bar_width)
-    plt.yticks(indexes, labels=labels)
-
-    st.pyplot(fig)
+    else:
+        labels = [kw[0].replace(' ', '\n') for kw in keywords]
+        scores = [kw[1] for kw in keywords]
+        #st.markdown(f"{keywords}")
+
+        # Топ 5 слов.
+        top = 5
+        top = min(len(labels), top)
+        st.markdown("Топ %d ключевых слов: **%s**" % (top, ', '.join(labels[0:5])))
+
+        # График важности слов.
+        fig, ax = plt.subplots(figsize=(8, len(labels)))
+        ax.set_title("95% самых важных ключевых слов")
+        ax.grid(color='#000000', alpha=0.15, linestyle='-', linewidth=1, which='major')
+        ax.grid(color='#000000', alpha=0.1, linestyle='-', linewidth=0.5, which='minor')
+
+        bar_width = 0.75
+        indexes = -np.arange(len(labels))
+        ax.barh(indexes, scores, bar_width)
+        plt.yticks(indexes, labels=labels)
+
+        st.pyplot(fig)
 else:
     st.markdown("Пожалуйста, введите что-нибудь.")