Manage error with fixed font

2023-12-26 22:59:57 -03:00 · 2023-12-26 22:59:57 -03:00 · d614022b59
parent 79bda1f639
commit d614022b59
1 changed files with 1 additions and 1 deletions
--- a/text_generation/generate_random_text.py
+++ b/text_generation/generate_random_text.py
@ -31,7 +31,7 @@ for i in range(NUM_IMAGES):
                                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,