diff --git a/index.html b/index.html index d363421..e8eeeb2 100644 --- a/index.html +++ b/index.html @@ -31,10 +31,6 @@ - - - - @@ -123,30 +119,122 @@
TODO
Python code
path = './models/onnx/t5-small'; +tokenizer = AutoTokenizer.from_pretrained(path) +model = AutoModelForSeq2SeqLM.from_pretrained(path) + +text = 'translate English to French: Hello, how are you?.' +input_ids = tokenizer(text).input_ids + +outputTokenIds = model.generate(input_ids); +outputText = tokenizer.decode(outputTokenIds, skip_special_tokens=True);
Corresponding JavaScript Code
let path = './models/onnx/t5-small'; +let tokenizer = await AutoTokenizer.from_pretrained(path) +let model = await AutoModelForSeq2SeqLM.from_pretrained(path) + +let text = 'translate English to French: Hello, how are you?.' +let input_ids = tokenizer(text).input_ids + +let outputTokenIds = await model.generate(input_ids); +let outputText = tokenizer.decode(outputTokenIds, skip_special_tokens=true);