fix #668

zyddnys · Jul 21, 2024 · 47bea86 · 47bea86
1 parent 6a00f74
commit 47bea86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/manga_translator/ocr/model_48px.py b/manga_translator/ocr/model_48px.py
@@ -63,7 +63,7 @@ async def _load(self, device: str):
     async def _unload(self):
         del self.model
 
-    async def _infer(self, image: np.ndarray, textlines: List[Quadrilateral], verbose: bool = False, ignore_bubble: int = 0) -> List[TextBlock]:
+    async def _infer(self, image: np.ndarray, textlines: List[Quadrilateral], args: dict, verbose: bool = False, ignore_bubble: int = 0) -> List[TextBlock]:
         text_height = 48
         max_chunk_size = 16