Update app.py

app.py

@@ -5,7 +5,7 @@ from sklearn.cluster import KMeans
 import re
 from io import BytesIO
 import tempfile
-from wordcloud import WordCloud
+from wordcloud import WordCloud, STOPWORDS
 import matplotlib.pyplot as plt
 from PIL import Image
 
@@ -74,8 +74,9 @@ def cluster_data(df, num_clusters):
 
 def generate_wordcloud(df):
     text = " ".join(df['texts'].tolist())
-    wordcloud = WordCloud(width=800, height=400, background_color='white', max_words=200, collocations=False).generate(text)
-    plt.figure(figsize=(10, 5))
+    stopwords = set(STOPWORDS)
+    wordcloud = WordCloud(width=1000, height=500, background_color='white', max_words=300, collocations=False, min_font_size=5, max_font_size=100, prefer_horizontal=1.0, stopwords=stopwords).generate(text)
+    plt.figure(figsize=(15, 7))
     plt.imshow(wordcloud, interpolation='bilinear')
     plt.axis('off')
     buf = BytesIO()