From cbb83dc5fd7ee50a8142750d69323f765e24228e Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Wed, 14 Dec 2022 19:27:12 +0800
Subject: [PATCH] Fix run.py

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 benchmark/run.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/benchmark/run.py b/benchmark/run.py
index 6672eee..2c2d5a3 100644
--- a/benchmark/run.py
+++ b/benchmark/run.py
@@ -119,6 +119,10 @@ elif args.format == 'onnx':
     if not os.path.exists(onnx_path):
         onnx_path = op.save_model(format='onnx')
 
+    sess = onnxruntime.InferenceSession(onnx_path,
+                                        providers=onnxruntime.get_available_providers())
+
+
     @towhee.register
     def run_onnx(img):
         img = img.convert('RGB')