Dabs commited on
Commit
d1115b0
·
1 Parent(s): 78bf5c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)