dicta-il
/

dictabert

Inference Endpoints

Model card Files Files and versions Community

Shaltiel commited on Aug 29, 2023

Commit

91e3018

•

1 Parent(s): 356be77

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -23,6 +23,7 @@ sentence = 'בשנת 1948 השלים אפרים קישון את [MASK] בפיס
 output = model(tokenizer.encode(sentence, return_tensors='pt'))
 # the [MASK] is the 7th token (including [CLS])
 top_2 = torch.topk(output.logits[0, 7, :], 2)[1]
 print('\n'.join(tokenizer.convert_ids_to_tokens(top_2))) # should print לימודיו / התמחותו

 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 לימודיו / התמחותו