Update README.md (#4)

- Update README.md (8603bc245ae768b7579ee1299672981e96e417ca)


Co-authored-by: Haoliang Tan <haoliangtan@users.noreply.huggingface.co>

README.md

@@ -63,7 +63,7 @@ You can use the raw model for image classification. See the [model hub](https://
                   transforms.ToTensor(),  
                   normalize])
   img_tensor = img_transformer(Image.open(image_path)).unsqueeze(0)
-  
+  img_tensor = torch.permute(img_tensor, (0, 2, 3, 1))
   so = onnxruntime.SessionOptions()
   ort_session = onnxruntime.InferenceSession(
           onnx_model, so,