From b2ee0cad418f1145000bc7233dee6081131e7992 Mon Sep 17 00:00:00 2001 From: PhanThanhTrung Date: Tue, 12 Mar 2024 14:49:17 +0700 Subject: [PATCH] update process_img parameter --- cope2n-ai-fi/api/sdsap_sbt/prediction_sbt.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/cope2n-ai-fi/api/sdsap_sbt/prediction_sbt.py b/cope2n-ai-fi/api/sdsap_sbt/prediction_sbt.py index c03b041..022b51e 100755 --- a/cope2n-ai-fi/api/sdsap_sbt/prediction_sbt.py +++ b/cope2n-ai-fi/api/sdsap_sbt/prediction_sbt.py @@ -38,7 +38,7 @@ def sbt_predict(image_url, engine) -> None: os.makedirs(save_dir, exist_ok = True) tmp_image_path = os.path.join(save_dir, f"{uuid.uuid4()}.jpg") cv2.imwrite(tmp_image_path, img) - outputs = process_img(img_path=tmp_image_path, + outputs = process_img(img=tmp_image_path, save_dir=save_dir, engine=engine, export_all=False, # False @@ -71,7 +71,6 @@ def predict(page_numb, image_url): """ sbt_result = sbt_predict(image_url, engine=sbt_engine) - print(sbt_result) output_dict = { "document_type": "invoice", "document_class": " ",