Add kwargs for rapidocr package.

SWHL · Oct 8, 2023 · f8b8d55 · f8b8d55
1 parent ceaee08
commit f8b8d55
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rapid_videocr/rapid_videocr.py b/rapid_videocr/rapid_videocr.py
@@ -27,6 +27,7 @@ def __init__(
         concat_batch: int = 10,
         out_format: str = "all",
         is_print_console: bool = False,
+        **kwargs,
     ) -> None:
         """Init
 
@@ -36,7 +37,7 @@ def __init__(
             out_format (str, optional): Output format of subtitle(srt, txt, all). Defaults to 'all'.
             is_print_console (bool, optional): Whether to print the subtitle results to console. 1 means to print results to console. Default is 0.
         """
-        self.rapid_ocr = RapidOCR(width_height_ratio=-1)
+        self.rapid_ocr = RapidOCR(width_height_ratio=-1, **kwargs)
         self.cropper = CropByProject()
 
         self.batch_size = concat_batch