awacke1 commited on
Commit
0c3b067
·
1 Parent(s): f733d00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -39,13 +39,10 @@ def display_top_words(content):
39
  text = ' '.join([elem.text for elem in soup.find_all(['p', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6'])])
40
  vectorizer = CountVectorizer()
41
  X = vectorizer.fit_transform([text])
42
- try:
43
- terms = vectorizer.get_feature_names()
44
- word_freq = X.toarray()[0]
45
- top_words = [terms[i] for i in word_freq.argsort()[-10:][::-1]]
46
- st.write(f"Top words: {', '.join(top_words)}")
47
- except:
48
- return
49
 
50
  def main():
51
  st.set_page_config(layout='wide')
@@ -60,7 +57,7 @@ def main():
60
  st.markdown(f"```{content.decode()}```")
61
 
62
  plot_word_frequencies(content)
63
- # display_top_words(content)
64
 
65
  if __name__ == '__main__':
66
  main()
 
39
  text = ' '.join([elem.text for elem in soup.find_all(['p', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6'])])
40
  vectorizer = CountVectorizer()
41
  X = vectorizer.fit_transform([text])
42
+ terms = vectorizer.get_feature_names()
43
+ word_freq = X.toarray()[0]
44
+ top_words = [terms[i] for i in word_freq.argsort()[-10:][::-1]]
45
+ st.write(f"Top words: {', '.join(top_words)}")
 
 
 
46
 
47
  def main():
48
  st.set_page_config(layout='wide')
 
57
  st.markdown(f"```{content.decode()}```")
58
 
59
  plot_word_frequencies(content)
60
+ display_top_words(content)
61
 
62
  if __name__ == '__main__':
63
  main()