phlippseitz
commited on
Commit
•
4338607
1
Parent(s):
c536d81
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -22,7 +22,7 @@ onnx_model = InferenceSession(onnx_path, providers=['CPUExecutionProvider'])
|
|
22 |
|
23 |
context = 'ONNX is an open format to represent models. The benefits of using ONNX include interoperability of frameworks and hardware optimization.'
|
24 |
question = 'What are advantages of ONNX?'
|
25 |
-
tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased')
|
26 |
|
27 |
inputs = tokenizer(question, context, padding=True, truncation=True, return_tensors='np')
|
28 |
inputs_int64 = {key: np.array(inputs[key], dtype=np.int64) for key in inputs}
|
|
|
22 |
|
23 |
context = 'ONNX is an open format to represent models. The benefits of using ONNX include interoperability of frameworks and hardware optimization.'
|
24 |
question = 'What are advantages of ONNX?'
|
25 |
+
tokenizer = AutoTokenizer.from_pretrained('UKP-SQuARE/bert-base-uncased-pf-drop-onnx')
|
26 |
|
27 |
inputs = tokenizer(question, context, padding=True, truncation=True, return_tensors='np')
|
28 |
inputs_int64 = {key: np.array(inputs[key], dtype=np.int64) for key in inputs}
|