Move dep

VikParuchuri · Feb 7, 2025 · d560369 · d560369
1 parent 2cf6a77
commit d560369
Show file tree

Hide file tree

Showing 2 changed files with 131 additions and 129 deletions.
diff --git a/benchmark/utils/textract.py b/benchmark/utils/textract.py
@@ -6,9 +6,7 @@
 from surya.input.processing import slice_bboxes_from_image
 from surya.recognition import RecognitionPredictor
 
-from textractor import Textractor
-
-def textract_ocr(extractor:Textractor, img):
+def textract_ocr(extractor, img):
     try:
         document = extractor.detect_document_text(file_source=img)
         return [line.text for line in document.lines]
@@ -17,6 +15,8 @@ def textract_ocr(extractor:Textractor, img):
         return [None]
 
 def textract_ocr_parallel(imgs, cpus=None):
+    from textractor import Textractor # Optional dependency
+
     extractor = Textractor(profile_name='default')
     parallel_cores = min(len(imgs), RecognitionPredictor().get_batch_size())
     if not cpus: