Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
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 |
-
|
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()
|