Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,8 @@ import gradio as gr
|
|
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 |
mask = np.array(input_img)
|
|
|
8 |
|
9 |
|
10 |
def create_wc(text, lang, custom_sw, input_img, color_rgb):
|
11 |
+
for csw in custom_sw:
|
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)
|