Update README.md

README.md

@@ -11,7 +11,7 @@ Known Issues:
 https://github.com/IAHLT/summarization_he
 
 
-```# !pip install bitsandbytes>=0.41.3 to quantize
+```# install bitsandbytes>=0.41.3 to quantize
 import torch
 from transformers import (
     AutoModelForCausalLM,
@@ -20,7 +20,7 @@ from transformers import (
 )
 
 
-def predict_text(text, tokenizer, model, num_beams=4, temperature=1, max_new_tokens=512):
+def summarize(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')
@@ -58,7 +58,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 
 text = 'טקסט לסיכום'
 
-predict_text(text, max_new_tokens=512, tokenizer=tokenizer, model=model)
+summarize(text, max_new_tokens=512, tokenizer=tokenizer, model=model)
 ```
 
 # Examples: