Merge pull request #18 from VikParuchuri/dev

Fix font bug
VikParuchuri · Feb 12, 2024 · 7e53a6d · 7e53a6d
2 parents 97f1432 + 0a0ad87
commit 7e53a6d
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "surya-ocr"
-version = "0.2.0"
+version = "0.2.1"
 description = "Document OCR models for multilingual text detection and recognition"
 authors = ["Vik Paruchuri <[email protected]>"]
 readme = "README.md"

diff --git a/static/images/benchmark_rec_chart.png b/static/images/benchmark_rec_chart.png
diff --git a/static/images/surya_rec_perf.png b/static/images/surya_rec_perf.png
diff --git a/surya/postprocessing/text.py b/surya/postprocessing/text.py
@@ -27,6 +27,7 @@ def draw_text_on_image(bboxes, texts, image_size=(1024, 1024), font_path=setting
 
         # Download font if it doesn't exist
         if not os.path.exists(font_path):
+            os.makedirs(os.path.dirname(font_path), exist_ok=True)
             with requests.get(settings.RECOGNITION_FONT_DL_PATH, stream=True) as r, open(font_path, 'wb') as f:
                 r.raise_for_status()
                 for chunk in r.iter_content(chunk_size=8192):