Spaces:

phamha
/

engineering-drawing

Running

Harry Pham commited on Apr 2

Commit

1023df1

1 Parent(s): d2abad3

update OCR

Files changed (1) hide show

src/inference.py CHANGED Viewed

@@ -725,5 +725,5 @@ def run_pipeline(image_path, output_dir="outputs",
 if __name__ == "__main__":
     import sys
     img = sys.argv[1] if len(sys.argv) > 1 else "test.jpg"
-    result, _ = run_pipeline(img, ocr_backend="surya")
     print(json.dumps(result, ensure_ascii=False, indent=2))

 if __name__ == "__main__":
     import sys
     img = sys.argv[1] if len(sys.argv) > 1 else "test.jpg"
+    result, _ = run_pipeline(img, ocr_backend="easyocr")
     print(json.dumps(result, ensure_ascii=False, indent=2))