Dabs commited on
Commit
78bf5c8
1 Parent(s): 129ca63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ def create_wc(text, lang, custom_sw, input_img, color_rgb):
11
  STOPWORDS = set(get_stop_words(lang))
12
  STOPWORDS.update(custom_sw.replace(" ", "").split(","))
13
  mask = np.array(input_img)
14
- wordcloud = WordCloud(background_color="rgba(0, 0, 0, 0)", mode="RGBA",mask=mask, width=1000, height=1500, stopwords=STOPWORDS).generate(text.lower())
 
15
  # wordcloud.recolor(colormap=colormap)
16
  wordcloud.recolor(color_func=get_single_color_func(f'rgb({color_rgb})'))
17
 
 
11
  STOPWORDS = set(get_stop_words(lang))
12
  STOPWORDS.update(custom_sw.replace(" ", "").split(","))
13
  mask = np.array(input_img)
14
+ text_dict = Counter(text.lower().split())
15
+ wordcloud = WordCloud(background_color="rgba(0, 0, 0, 0)", mode="RGBA",mask=mask, width=1000, height=1500, stopwords=STOPWORDS).generate_from_frequencies(text_dict)
16
  # wordcloud.recolor(colormap=colormap)
17
  wordcloud.recolor(color_func=get_single_color_func(f'rgb({color_rgb})'))
18