Update app.py

app.py

@@ -10,10 +10,10 @@ import gradio as gr
 def create_wc(text, lang, custom_sw, input_img, color_rgb):
     STOPWORDS = set(get_stop_words(lang))
     STOPWORDS.update(custom_sw.replace(" ", "").split(","))
-    for csw in STOPWORDS:
-        text = text.replace(csw, "")
+    words = text.lower()split(" ")
+    words = [word for word in word_list if word not in STOPWORDS]
     mask = np.array(input_img)
-    text_dict = Counter(text.lower().split())
+    text_dict = Counter(words)
     wordcloud = WordCloud(background_color="rgba(0, 0, 0, 0)", mode="RGBA",mask=mask, width=1000, height=1500, stopwords=STOPWORDS).generate_from_frequencies(text_dict)
     # wordcloud.recolor(colormap=colormap)
     wordcloud.recolor(color_func=get_single_color_func(f'rgb({color_rgb})'))