Update README.md

README.md

@@ -66,7 +66,7 @@ text = "Your text goes here..."
 
 # If you want to use CPU
 input_ids = tokenizer(text, return_tensors="pt", truncation=True).input_ids
-# Ir you want to use GPU
+# If you want to use GPU
 input_ids = tokenizer(text, return_tensors="pt", truncation=True).input_ids.cuda()
 # Adjust max_new_tokens based on size. This is set up for articles of text
 outputs = model.generate(input_ids=input_ids, max_new_tokens=120, do_sample=False)
@@ -74,8 +74,6 @@ outputs = model.generate(input_ids=input_ids, max_new_tokens=120, do_sample=Fals
 print(f"input sentence: {sample['article']}\n{'---'* 20}")
 print(f"summary:\n{tokenizer.batch_decode(outputs.detach().cpu().numpy(), skip_special_tokens=True)[0]}")
 
-
-
 ```