Update app.py
Browse files
app.py
CHANGED
|
@@ -8,10 +8,10 @@ import gradio as gr
|
|
| 8 |
|
| 9 |
|
| 10 |
def create_wc(text, lang, custom_sw, input_img, color_rgb):
|
| 11 |
-
for csw in custom_sw.split(","):
|
| 12 |
-
text = text.replace(csw, "")
|
| 13 |
STOPWORDS = set(get_stop_words(lang))
|
| 14 |
STOPWORDS.update(custom_sw.replace(" ", "").split(","))
|
|
|
|
|
|
|
| 15 |
mask = np.array(input_img)
|
| 16 |
text_dict = Counter(text.lower().split())
|
| 17 |
wordcloud = WordCloud(background_color="rgba(0, 0, 0, 0)", mode="RGBA",mask=mask, width=1000, height=1500, stopwords=STOPWORDS).generate_from_frequencies(text_dict)
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
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 |
+
for csw in STOPWORDS:
|
| 14 |
+
text = text.replace(csw, "")
|
| 15 |
mask = np.array(input_img)
|
| 16 |
text_dict = Counter(text.lower().split())
|
| 17 |
wordcloud = WordCloud(background_color="rgba(0, 0, 0, 0)", mode="RGBA",mask=mask, width=1000, height=1500, stopwords=STOPWORDS).generate_from_frequencies(text_dict)
|