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