madhavgohel
/

bert-token-onnx

Model card Files Files and versions

madhavgohel commited on 20 days ago

Commit

fb83494

·

verified ·

1 Parent(s): 3480267

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -32,7 +32,7 @@ from transformers import AutoTokenizer
 from optimum.onnxruntime import ORTModelForTokenClassification
 import torch
-model = ORTModelForTokenClassification.from_pretrained("your-username/bert-token-onnx", file_name="model.onnx")
 tokenizer = AutoTokenizer.from_pretrained("your-username/bert-token-onnx")
 text = "The server will go down at midnight for maintenance."

 from optimum.onnxruntime import ORTModelForTokenClassification
 import torch
+model = ORTModelForTokenClassification.from_pretrained("madhavgohel/bert-token-onnx", file_name="model.onnx")
 tokenizer = AutoTokenizer.from_pretrained("your-username/bert-token-onnx")
 text = "The server will go down at midnight for maintenance."