diff --git a/tools/onnx_subgraph/extract_onnx.py b/tools/onnx_subgraph/extract_onnx.py
index 4204901155b..76133885b2f 100644
--- a/tools/onnx_subgraph/extract_onnx.py
+++ b/tools/onnx_subgraph/extract_onnx.py
@@ -36,8 +36,13 @@ def split_onnx_ios(instrfile, input_path, out_folder='subgraphs/'):
         model.graph.value_info.append(output)
     onnx.save(model, input_path)
 
-    f1 = open(instrfile, "r")
-    lines = f1.readlines()
+    try:
+        with open(instrfile, "r") as f1:
+            lines = f1.readlines()
+    except Exception as e:
+        print(e)
+        raise
+
     cpu_count = 0
     npu_count = 0
     count = 0
@@ -61,8 +66,6 @@ def split_onnx_ios(instrfile, input_path, out_folder='subgraphs/'):
             print("succeed", count)
             count = count + 1
 
-    f1.close()
-
 
 if __name__ == "__main__":
     arg_parser = argparse.ArgumentParser()