Update app.py

app.py

@@ -30,7 +30,7 @@ def get_text_from_image(img):
     return bg
 
 
-def split_text_to_rows(text, n=4):
+def split_text_to_rows(text, n):
     # https://stackoverflow.com/a/6187258
     l = text.split()
     return [' '.join(l[x:x+n]) for x in range(0, len(l), n)]
@@ -64,7 +64,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)
+        texts = split_text_to_rows(text, WORDS_IN_ROW)
     else:
         texts = [row.strip() for row in text.split(NEW_LINE_SYMB)]
 
@@ -103,6 +103,7 @@ GENERATOR = ImgGenerator(
 BATCH_SIZE = 3
 ROW_WIDTH = 0.7
 LEFT_PAD = 10
+WORDS_IN_ROW = 4
 NEW_LINE_SYMB = '{n}'
 
 gr.Interface(