Update worker.py

Added model version config
DrRSatzteil · Jan 6, 2024 · 05f0bb6 · 05f0bb6
1 parent 641ccc0
commit 05f0bb6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mayanmindee/worker.py b/mayanmindee/worker.py
@@ -245,14 +245,15 @@ def process_custom(document_id: int, document_type: str, synchronous: False) ->
 
     account_name = apis[document_type]["account"]
     endpoint_name = apis[document_type]["endpoint"]
+    model_version = apis[document_type]["model"]
 
     m = get_mayan()
     document, pdf_bytes = load_document(m, document_id)
 
     file = cut_pages(pdf_bytes, pagelimit=apis[document_type]["pagelimit"])
 
     parsed_doc = ocr_custom(
-        file, str(document_id) + ".pdf", account_name, endpoint_name, synchronous
+        file, str(document_id) + ".pdf", account_name, endpoint_name, model_version, synchronous
     )
 
     if "storeocr" in apis[document_type]: