🧪 Add tests for text detection

hezarai · Jul 5, 2024 · 2bdf213 · 2bdf213
1 parent cfcfd75
commit 2bdf213
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/tests/samples/text_detection_example.png b/tests/samples/text_detection_example.png
diff --git a/tests/test_models_inference.py b/tests/test_models_inference.py
@@ -33,6 +33,13 @@
  "output_type_within_batch": dict,
  "required_output_keys": {"text", "score"}
  },
+ "text_detection": {
+ "path": "hezarai/CRAFT",
+ "inputs": {"type": "file", "value": "samples/text_detection_example.png"},
+ "predict_kwargs": {},
+ "output_type_within_batch": dict,
+ "required_output_keys": {"boxes"}
+ },
  "ocr": {
  "path": "hezarai/crnn-fa-printed-96-long",
  "inputs": {"type": "file", "value": "samples/ocr_example.jpg"},