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)
|