Update README.md

README.md

@@ -60,7 +60,7 @@ model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, tru
 tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path,trust_remote_code=True)
 model = PeftModel.from_pretrained(model, peft_model_id)
 
-#Put from dataset
+#from dataset
 inputs = tokenizer(f'''{formatted_dataset['text'][120]}''', return_tensors="pt", return_attention_mask=False)
 outputs = model.generate(**inputs,max_new_tokens=50, pad_token_id = tokenizer.eos_token_id, eos_token_id = tokenizer.eos_token_id)
 text = tokenizer.batch_decode(outputs)[0]
@@ -68,7 +68,7 @@ print(text)
 ```
 
 ```python
-#Put as string
+#as string
 inputs = tokenizer(f'''What is the title of this paper? ...[your pdf as text]..\n\nAnswer: ''', return_tensors="pt", return_attention_mask=False)
 outputs = model.generate(**inputs,max_new_tokens=50, pad_token_id = tokenizer.eos_token_id, eos_token_id = tokenizer.eos_token_id)
 text = tokenizer.batch_decode(outputs)[0]