vanessbut commited on
Commit
8f5b334
·
1 Parent(s): d100f15

Добавлена гистограма.

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -75,7 +75,9 @@ text = preprocess([title + ". " + abstract])[0]
75
 
76
  if not text is None and len(text) > 0:
77
  #keywords = get_candidates(text, main_nlp)
78
- keywords = np.array(get_keywords(text, main_nlp, main_model, main_tokenizer))
 
 
79
  #st.markdown(f"{keywords}")
80
 
81
  # График важности слов.
@@ -83,9 +85,9 @@ if not text is None and len(text) > 0:
83
  ax.set_title("Ключевые слова в порядке важности")
84
 
85
  bar_width = 0.35
86
- indexes = np.arange(keywords.shape[0])
87
- ax.barh(indexes, keywords[:,1], bar_width)
88
- plt.yticks(indexes, keywords[:,0])
89
 
90
  st.pyplot(fig)
91
  else:
 
75
 
76
  if not text is None and len(text) > 0:
77
  #keywords = get_candidates(text, main_nlp)
78
+ keywords = get_keywords(text, main_nlp, main_model, main_tokenizer)
79
+ labels = [kw[0] for kw in keywords]
80
+ scores = [kw[1] for kw in keywords]
81
  #st.markdown(f"{keywords}")
82
 
83
  # График важности слов.
 
85
  ax.set_title("Ключевые слова в порядке важности")
86
 
87
  bar_width = 0.35
88
+ indexes = np.arange(len(labels))
89
+ ax.barh(indexes, scores, bar_width)
90
+ plt.yticks(indexes, labels=labels)
91
 
92
  st.pyplot(fig)
93
  else: