Support subword and word language models

Closes #199 (closed)

Ref https://redmine.teklia.com/issues/4941

dan/ocr/predict/prediction.py

@@ -370,11 +370,10 @@ def process_batch(
 
         # Return LM results
         if use_language_model:
-            result["language_model"] = {}
-            result["language_model"]["text"] = prediction["language_model"]["text"][idx]
-            result["language_model"]["confidence"] = prediction["language_model"][
-                "confidence"
-            ][idx]
+            result["language_model"] = {
+                "text": prediction["language_model"]["text"][idx],
+                "confidence": prediction["language_model"]["confidence"][idx],
+            }
 
         # Return extracted objects (coordinates, text, confidence)
         if predict_objects: