ppisljar
/

slo_g2p_norm_byt5

Model card Files Files and versions

ppisljar commited on Feb 21, 2024

Commit

665ef44

·

verified ·

1 Parent(s): e2dd5f3

Update infer.py

Files changed (1) hide show

infer.py +1 -1

infer.py CHANGED Viewed

@@ -17,7 +17,7 @@ tokenizer = AutoTokenizer.from_pretrained('google/byt5-small')
 sentence = "Kupil sem bicikel in mu zamenjal stol.".lower()
-ort_session = onnxruntime.InferenceSession("g2p_t5.onnx", providers=["CPUExecutionProvider"])
 input_ids = [sentence]
 input_encoding = tokenizer(
     input_ids, padding='longest', max_length=512, truncation=True, return_tensors='pt',

 sentence = "Kupil sem bicikel in mu zamenjal stol.".lower()
+ort_session = onnxruntime.InferenceSession("g2p_norm_t5.onnx", providers=["CPUExecutionProvider"])
 input_ids = [sentence]
 input_encoding = tokenizer(
     input_ids, padding='longest', max_length=512, truncation=True, return_tensors='pt',