Update app.py

app.py

@@ -7,9 +7,16 @@ from collections import Counter
 import gradio as gr
 
 
+def replace(string, char):
+    pattern = char + '{2,}'
+    string = re.sub(pattern, char, string)
+    return string
+    
 def create_wc(text, lang, custom_sw, input_img, color_rgb):
     STOPWORDS = set(get_stop_words(lang))
     STOPWORDS.update(custom_sw.replace(" ", "").split(","))
+    text = replace(text, "o")
+    text = replace(text, "a")
     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())
     # wordcloud.recolor(colormap=colormap)