From 694ff2e8291526889c1aff071d00fbc28ff07737 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maciej=20Budy=C5=9B?= <kha-white@mail.com>
Date: Thu, 3 Nov 2022 21:33:10 +0100
Subject: [PATCH] pass max_length to generate method

---
 manga_ocr/ocr.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/manga_ocr/ocr.py b/manga_ocr/ocr.py
index 8710353..ee188b4 100644
--- a/manga_ocr/ocr.py
+++ b/manga_ocr/ocr.py
@@ -39,7 +39,7 @@ class MangaOcr:
         img = img.convert('L').convert('RGB')
 
         x = self._preprocess(img)
-        x = self.model.generate(x[None].to(self.model.device))[0].cpu()
+        x = self.model.generate(x[None].to(self.model.device), max_length=300)[0].cpu()
         x = self.tokenizer.decode(x, skip_special_tokens=True)
         x = post_process(x)
         return x