Spaces:

Dabs
/

wordcloud

Runtime error

Dabs commited on Feb 18, 2022

Commit

abd411e

•

1 Parent(s): 99adff3

lang

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import gradio as gr
 def create_wc(text, lang, custom_sw, input_img, color_rgb):
-    STOPWORDS = set(get_stop_words('en'))
     STOPWORDS.update(custom_sw.replace(" ", "").split(","))
     mask = np.array(input_img)
     wordcloud = WordCloud(background_color="rgba(0, 0, 0, 0)", mode="RGBA",mask=mask, width=1000, height=1500, stopwords=STOPWORDS).generate(text.lower())

 def create_wc(text, lang, custom_sw, input_img, color_rgb):
+    STOPWORDS = set(get_stop_words(lang))
     STOPWORDS.update(custom_sw.replace(" ", "").split(","))
     mask = np.array(input_img)
     wordcloud = WordCloud(background_color="rgba(0, 0, 0, 0)", mode="RGBA",mask=mask, width=1000, height=1500, stopwords=STOPWORDS).generate(text.lower())