fix: Fixed API

mindee · Sep 20, 2021 · 6a572a0 · 6a572a0
1 parent 9e267e8
commit 6a572a0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/api/app/routes/detection.py b/api/app/routes/detection.py
@@ -18,5 +18,5 @@
 async def text_detection(file: UploadFile = File(...)):
     """Runs DocTR text detection model to analyze the input"""
     img = decode_img_as_tensor(file.file.read())
-    boxes, _ = det_predictor([img], training=False)[0]
+    boxes, _ = det_predictor([img])[0]
     return [DetectionOut(box=box.tolist()) for box in boxes[:, :-1]]
diff --git a/api/app/routes/ocr.py b/api/app/routes/ocr.py
@@ -18,7 +18,7 @@
 async def perform_ocr(file: UploadFile = File(...)):
     """Runs DocTR OCR model to analyze the input"""
     img = decode_img_as_tensor(file.file.read())
-    out = predictor([img], training=False)
+    out = predictor([img])
 
     return [OCROut(box=(*word.geometry[0], *word.geometry[1]), value=word.value)
             for word in out.pages[0].blocks[0].lines[0].words]
diff --git a/api/app/routes/recognition.py b/api/app/routes/recognition.py
@@ -17,5 +17,5 @@
 async def text_recognition(file: UploadFile = File(...)):
     """Runs DocTR text recognition model to analyze the input"""
     img = decode_img_as_tensor(file.file.read())
-    out = reco_predictor([img], training=False)
+    out = reco_predictor([img])
     return RecognitionOut(value=out[0][0])