Add 0.5.0 features

app.py

@@ -44,6 +44,8 @@ def load_predictor(
     load_in_8_bit: bool,
     bin_thresh: float,
     box_thresh: float,
+    disable_crop_orientation: bool = False,
+    disable_page_orientation: bool = False,
 ) -> OCRPredictor:
     """Load a predictor from doctr.models
 
@@ -52,6 +54,8 @@ def load_predictor(
         det_arch: detection architecture
         reco_arch: recognition architecture
         assume_straight_pages: whether to assume straight pages or not
+        disable_crop_orientation: whether to disable crop orientation or not
+        disable_page_orientation: whether to disable page orientation or not
         straighten_pages: whether to straighten rotated pages or not
         detect_language: whether to detect the language of the text
         load_in_8_bit: whether to load the image in 8 bit mode
@@ -71,6 +75,8 @@ def load_predictor(
         load_in_8_bit=load_in_8_bit,
         export_as_straight_boxes=straighten_pages,
         detect_orientation=not assume_straight_pages,
+        disable_crop_orientation=disable_crop_orientation,
+        disable_page_orientation=disable_page_orientation,
     )
     predictor.det_predictor.model.postprocessor.bin_thresh = bin_thresh
     predictor.det_predictor.model.postprocessor.box_thresh = box_thresh
@@ -122,6 +128,8 @@ def analyze_page(
     det_arch: str,
     reco_arch: str,
     assume_straight_pages: bool,
+    disable_crop_orientation: bool,
+    disable_page_orientation: bool,
     straighten_pages: bool,
     detect_language: bool,
     load_in_8_bit: bool,
@@ -137,6 +145,8 @@ def analyze_page(
         det_arch: detection architecture
         reco_arch: recognition architecture
         assume_straight_pages: whether to assume straight pages or not
+        disable_crop_orientation: whether to disable crop orientation or not
+        disable_page_orientation: whether to disable page orientation or not
         straighten_pages: whether to straighten rotated pages or not
         detect_language: whether to detect the language of the text
         load_in_8_bit: whether to load the image in 8 bit mode
@@ -167,6 +177,8 @@ def analyze_page(
         load_in_8_bit,
         bin_thresh,
         box_thresh,
+        disable_crop_orientation=disable_crop_orientation,
+        disable_page_orientation=disable_page_orientation,
     )
 
     seg_map = forward_image(predictor, page)
@@ -215,6 +227,8 @@ with gr.Blocks(fill_height=True) as demo:
             det_model = gr.Dropdown(choices=DET_ARCHS, value=DET_ARCHS[0], label="Text detection model")
             reco_model = gr.Dropdown(choices=RECO_ARCHS, value=RECO_ARCHS[0], label="Text recognition model")
             assume_straight = gr.Checkbox(value=True, label="Assume straight pages")
+            disable_crop_orientation = gr.Checkbox(value=False, label="Disable crop orientation")
+            disable_page_orientation = gr.Checkbox(value=False, label="Disable page orientation")
             straighten = gr.Checkbox(value=False, label="Straighten pages")
             det_language = gr.Checkbox(value=False, label="Detect language")
             load_in_8_bit = gr.Checkbox(value=False, label="Load 8-bit quantized models")
@@ -242,6 +256,8 @@ with gr.Blocks(fill_height=True) as demo:
             det_model,
             reco_model,
             assume_straight,
+            disable_crop_orientation,
+            disable_page_orientation,
             straighten,
             det_language,
             load_in_8_bit,