.

src/eynollah/eynollah_ocr.py

@@ -72,7 +72,7 @@ class Eynollah_ocr:
             self.model_zoo.load_model('ocr', '')
             self.model_zoo.load_model('num_to_char')
             self.model_zoo.load_model('characters')
-            self.end_character = len(self.model_zoo.get('characters', list)) + 2
+            self.end_character = len(self.model_zoo.get('characters')) + 2
 
     @property
     def device(self):

src/eynollah/model_zoo/model_zoo.py

@@ -197,9 +197,12 @@ class EynollahModelZoo:
         return model
 
     def get(self, model_category: str) -> Predictor:
-        if model_category not in self._loaded:
-            raise ValueError(f'Model "{model_category}" not previously loaded with "load_model(..)"')
+        # if model_category not in self._loaded:
+        #     raise ValueError(f'Model "{model_category}" not previously loaded with "load_model(..)"')
+        if model_category in self._loaded:
             return self._loaded[model_category]
+        else:
+            return self.load_model(model_category)
 
     def _load_ocr_model(self, variant: str) -> AnyModel:
         """

src/eynollah/ocrd-tool.json

@@ -162,7 +162,6 @@
           "version_range": "< v0.7.0"
         }
       ]
-    }
     },
     "ocrd-eynollah-recognize": {
       "executable": "ocrd-eynollah-recognize",
@@ -170,6 +169,7 @@
       "steps": ["recognition/text-recognition"],
       "input_file_grp_cardinality": 1,
       "output_file_grp_cardinality": 1,
+      "description": "Recognize text with eynollah (CNN/RNN or Transformer)",
       "parameters": {
         "models": {
           "type": "string",
@@ -188,6 +188,12 @@
           "type": "boolean",
           "description": "Whether to use (much more resource-intensive) transformer model",
           "default": false
+        },
+	"batch_size": {
+	  "type": "number",
+	  "format": "integer",
+	  "description": "Batch size, leave as 0 for builtin default (2 for CNN/RNN, 2 for TrOCR)",
+	  "default": 0
 	}
       },
       "resources": [
@@ -202,3 +208,4 @@
       ]
     }
   }
+}

src/eynollah/ocrd_cli_recognize.py

@@ -34,8 +34,8 @@ class EynollahRecognizeProcessor(Processor):
             model_zoo=model_zoo,
             tr_ocr=self.parameter['tr_ocr'],
             do_not_mask_with_textline_contour=self.parameter['do_not_mask_with_textline_contour'],
-            batch_size=self.parameter['batch_size'],
-            min_conf_value_of_textline_text=self.parameter['min_conf_value_of_textline_text'])
+            batch_size=self.parameter['batch_size'] if self.parameter['batch_size'] >= 0 else 2 if self.parameter['tr_ocr'] else 8,
+            min_conf_value_of_textline_text=0)
 
     # FIXME: This is just a proof-of-concept, very inefficient and non-conformant
     # TODO: OCR writing should use PAGE API once result dataclass mechanism is settled,