Update app.py

app.py

@@ -220,6 +220,18 @@ with tab1:
         df = df[['content','Class']]
         
 
+        # Word Cloud Visualization
+        def create_wordcloud(text_data):
+            text = ' '.join(text_data)
+            wordcloud = WordCloud(width=800, height=400).generate(text)
+            plt.figure(figsize=(10, 5))
+            plt.imshow(wordcloud, interpolation='bilinear')
+            plt.axis('off')
+            st.pyplot(plt)
+
+        st.subheader("Word Cloud of News Content")
+        create_wordcloud(df['content'])
+
 
         #show Classification Results
         st.subheader("Classification Results")