Update README.md

README.md

@@ -46,20 +46,19 @@ model = AutoModelForMaskedLM.from_pretrained("GiliGold/Knesset-DictaBERT")
 
 model.eval()
 
-sentence = "הכנסת היא הרשות [MASK] של מדינת ישראל."
+sentence = "יש לנו [MASK] על זה בשבוע הבא"
 
 # Tokenize the input sentence and get predictions
 inputs = tokenizer.encode(sentence, return_tensors='pt')
 output = model(inputs)
 
-# The [MASK] token is the 5th token in the sentence (including [CLS])
-mask_token_index = 5
+mask_token_index = 3
 top_2_tokens = torch.topk(output.logits[0, mask_token_index, :], 2)[1]
 
 # Convert token IDs to tokens and print them
 print('\n'.join(tokenizer.convert_ids_to_tokens(top_2_tokens)))
 
-# Example output: המבצעת / המחוקקת
+# Example output: ישיבה / דיון
 
 ```