dicta-il
/

dictabert-tiny

Inference Endpoints

Model card Files Files and versions Community

Shaltiel commited on Mar 4

Commit

d6c74d0

•

1 Parent(s): c8cfd1d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ output = model(tokenizer.encode(sentence, return_tensors='pt'))
 # the [MASK] is the 7th token (including [CLS])
 import torch
 top_2 = torch.topk(output.logits[0, 7, :], 2)[1]
-print('\n'.join(tokenizer.convert_ids_to_tokens(top_2))) # should print לימודיו / מחקריו
 ```

 # the [MASK] is the 7th token (including [CLS])
 import torch
 top_2 = torch.topk(output.logits[0, 7, :], 2)[1]
+print('\n'.join(tokenizer.convert_ids_to_tokens(top_2))) # should print עבודתו / התמחותו
 ```