Update README.md

README.md

@@ -43,9 +43,7 @@ import torch
 
 tokenizer = AutoTokenizer.from_pretrained("GiliGold/Knesset-DictaBERT")
 model = AutoModelForMaskedLM.from_pretrained("GiliGold/Knesset-DictaBERT")
-
 model.eval()
-
 sentence = "יש לנו [MASK] על זה בשבוע הבא"
 
 # Tokenize the input sentence and get predictions
@@ -59,7 +57,6 @@ top_2_tokens = torch.topk(output.logits[0, mask_token_index, :], 2)[1]
 print('\n'.join(tokenizer.convert_ids_to_tokens(top_2_tokens)))
 
 # Example output: ישיבה / דיון
-
 ```
 
 ## Evaluation