diff options
Diffstat (limited to 'docs/models/word_level_pretrained.md')
-rw-r--r-- | docs/models/word_level_pretrained.md | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/docs/models/word_level_pretrained.md b/docs/models/word_level_pretrained.md index b173c6c..5c7e016 100644 --- a/docs/models/word_level_pretrained.md +++ b/docs/models/word_level_pretrained.md @@ -18,15 +18,16 @@ We have released several pre-trained TransQuest models on word-level quality est | \*-\* | Any | Any | MicroTransQuest | | You can load any of the above models with the below code. The full notebook is available [here.](https://colab.research.google.com/drive/1fslfFoQnspdv2Do5hfwwkbEwow-CDIxD?usp=sharing) +Let's consider loading microtransquest-en_lv-pharmaceutical-nmt model. ```python from transquest.algo.word_level.microtransquest.run_model import MicroTransQuestModel import torch model = MicroTransQuestModel("xlmroberta", "TransQuest/microtransquest-en_lv-pharmaceutical-nmt", labels=["OK", "BAD"], use_cuda=torch.cuda.is_available()) -``` - +source_tags, target_tags = model.predict([["if not , you may not be protected against the diseases . ", "ja tā nav , Jūs varat nepasargāt no slimībām . "]]) +``` !!! note |