Update README.md

README.md

@@ -21,6 +21,7 @@ from transformers import (
 
 
 def predict_text(text, tokenizer, model, num_beams=4, temperature=1, max_new_tokens=512):
+    # This text template is important.
     inputs = tokenizer(f'{text}\n### סיכום:', return_tensors="pt")
     in_data = inputs.input_ids.to('cuda')
     output_ids = model.generate(input_ids=in_data, num_beams=num_beams, max_new_tokens = max_new_tokens, do_sample=True, early_stopping=True, use_cache = True, temperature=temperature, eos_token_id=tokenizer.eos_token_id)