Update README.md

README.md

@@ -14,7 +14,7 @@ from urllib.request import urlopen
 
 import einops
 import numpy as np
-import onnxruntime
+import onnxruntime as ort
 from PIL import Image
 
 def softmax(x):
@@ -24,8 +24,8 @@ def softmax(x):
 IMG_URL = 'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
 IN1K_CLASSES_URL = 'https://storage.googleapis.com/bit_models/ilsvrc2012_wordnet_lemmas.txt'
 
-session = onnxruntime.InferenceSession('eva02_tiny_patch14_336.mim_in22k_ft_in1k.ort')
-# session = onnxruntime.InferenceSession('eva02_tiny_patch14_336.mim_in22k_ft_in1k.onnx')
+session = ort.InferenceSession('eva02_tiny_patch14_336.mim_in22k_ft_in1k.ort')
+# session = ort.InferenceSession('eva02_tiny_patch14_336.mim_in22k_ft_in1k.onnx')
 
 labels = urlopen(IN1K_CLASSES_URL).read().decode().splitlines()
 img = np.array(
@@ -43,7 +43,7 @@ img = einops.rearrange(img, 'h w c -> 1 c h w').astype(np.float32)
 out = session.run(None, {session.get_inputs()[0].name: img})
 
 out = softmax(out[0][0])
-topk = np.argsort(out)[::-1][:5] 
+topk = np.argsort(out)[::-1][:5]
 for i in topk:
     print(f'{out[i]:.2f}', labels[i])
 ```