rectified onnx inference code

README.md

@@ -106,7 +106,7 @@ predicted_class_id = logits.argmax().item()
 loaded_model.config.id2label[predicted_class_id]
 ```
 
-Optimum with ONNX
+### Optimum with ONNX inference
 
 Loading the model requires the 🤗 Optimum library installed.
 ```shell
@@ -115,12 +115,12 @@ pip install transformers optimum[onnxruntime] optimum
 
 ```python
 model_path = "philomath-1209/programming-language-identification"
-
+import torch
 from transformers import pipeline, AutoTokenizer
 from optimum.onnxruntime import ORTModelForSequenceClassification
 
-tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = ORTModelForSequenceClassification.from_pretrained(model_path, export=True)
+tokenizer = AutoTokenizer.from_pretrained(model_path, subfolder="onnx")
+model = ORTModelForSequenceClassification.from_pretrained(model_path, export=False, subfolder="onnx")
 
 text = """
   PROGRAM Triangle
@@ -141,9 +141,10 @@ text = """
     END FUNCTION Area
 
 """
-inputs = loaded_tokenizer(text, return_tensors="pt",truncation=True)
+inputs = tokenizer(text, return_tensors="pt",truncation=True)
 with torch.no_grad():
-    logits = loaded_model(**inputs).logits
+    logits = model(**inputs).logits
 predicted_class_id = logits.argmax().item()
-loaded_model.config.id2label[predicted_class_id]
+model.config.id2label[predicted_class_id]
+
 ```