From 02278c2d67f49dc2a3525de7bf74bfe6392b2a04 Mon Sep 17 00:00:00 2001
From: SWHL <liekkaskono@163.com>
Date: Fri, 30 Jun 2023 16:14:42 +0800
Subject: [PATCH] Update OrtInferSession tip when meets error.

---
 python/rapidocr_onnxruntime/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/rapidocr_onnxruntime/utils.py b/python/rapidocr_onnxruntime/utils.py
index 62e062a..0e5e4d0 100644
--- a/python/rapidocr_onnxruntime/utils.py
+++ b/python/rapidocr_onnxruntime/utils.py
@@ -2,6 +2,7 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 import argparse
+import traceback
 import warnings
 from io import BytesIO
 from pathlib import Path
@@ -71,7 +72,8 @@ class OrtInferSession:
         try:
             return self.session.run(self.get_output_names(), input_dict)
         except Exception as e:
-            raise ONNXRuntimeError("ONNXRuntime inference failed.") from e
+            error_info = traceback.format_exc()
+            raise ONNXRuntimeError(error_info) from e
 
     def get_input_names(
         self,