Spaces:
Sleeping
Sleeping
Harry Pham commited on
Commit ·
d6af665
1
Parent(s): 50d1f57
update OCR
Browse files- src/inference.py +1 -1
src/inference.py
CHANGED
|
@@ -1306,6 +1306,6 @@ def run_pipeline(image_path, output_dir="outputs",
|
|
| 1306 |
if __name__ == "__main__":
|
| 1307 |
import sys
|
| 1308 |
img = sys.argv[1] if len(sys.argv) > 1 else "test.jpg"
|
| 1309 |
-
backend = sys.argv[2] if len(sys.argv) > 2 else "
|
| 1310 |
result, _ = run_pipeline(img, ocr_backend=backend)
|
| 1311 |
print(json.dumps(result, ensure_ascii=False, indent=2))
|
|
|
|
| 1306 |
if __name__ == "__main__":
|
| 1307 |
import sys
|
| 1308 |
img = sys.argv[1] if len(sys.argv) > 1 else "test.jpg"
|
| 1309 |
+
backend = sys.argv[2] if len(sys.argv) > 2 else "easyocr"
|
| 1310 |
result, _ = run_pipeline(img, ocr_backend=backend)
|
| 1311 |
print(json.dumps(result, ensure_ascii=False, indent=2))
|