Manage error with fixed font

VJean-dev · Dec 27, 2023 · d614022 · d614022
1 parent 79bda1f
commit d614022
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/text_generation/generate_random_text.py b/text_generation/generate_random_text.py
@@ -31,7 +31,7 @@
                                 string.digits, k=NUM_CHARACTERS))
 
     text_color = random.choices(["black","white"], weights=(70, 30), k=1)[0]
-    background_color = random.choices(["black","white"], weights=(30, 70), k=1)[0]
+    background_color = "black"*(text_color=="white") + "white"*(text_color=="black")
 
     generate_random_txt_img(text, 
                             IMG_SHAPE,