Update app.py

app.py

@@ -36,6 +36,13 @@ def split_text_to_rows(text, n):
     return [' '.join(l[x:x+n]) for x in range(0, len(l), n)]
 
 
+def split_text_to_rows_by_chars(text, n):
+    list_of_rows = []
+    for i in range(0, len(a), n):
+        list_of_rows.append(a[i:n+i].strip())
+    return list_of_rows  
+
+
 def remove_right_padding(img, len_text, char_w=32):
     # char_w for a standard ScrabbleGAN char width
     return img[:, :len_text*char_w]
@@ -64,7 +71,7 @@ def get_canvas_size(images, row_width, left_pad):
 
 def predict(text):
     if text.find(NEW_LINE_SYMB) == -1:
-        texts = split_text_to_rows(text, WORDS_IN_ROW)
+        texts = split_text_to_rows_by_chars(text, CHARS_IN_ROW)
     else:
         texts = [row.strip() for row in text.split(NEW_LINE_SYMB)]
 
@@ -104,6 +111,7 @@ BATCH_SIZE = 3
 ROW_WIDTH = 0.7
 LEFT_PAD = 10
 WORDS_IN_ROW = 4
+CHARS_IN_ROW = 40
 NEW_LINE_SYMB = '{n}'
 
 gr.Interface(