Update app.py

app.py

@@ -39,10 +39,13 @@ def display_top_words(content):
     text = ' '.join([elem.text for elem in soup.find_all(['p', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6'])])
     vectorizer = CountVectorizer()
     X = vectorizer.fit_transform([text])
-    terms = vectorizer.get_feature_names()
-    word_freq = X.toarray()[0]
-    top_words = [terms[i] for i in word_freq.argsort()[-10:][::-1]]
-    st.write(f"Top words: {', '.join(top_words)}")
+    try:
+        terms = vectorizer.get_feature_names()
+        word_freq = X.toarray()[0]
+        top_words = [terms[i] for i in word_freq.argsort()[-10:][::-1]]
+        st.write(f"Top words: {', '.join(top_words)}")
+    except:
+        return
 
 def main():
     st.set_page_config(layout='wide')