Update README.md

README.md

@@ -50,7 +50,13 @@ def calculate_sum(a, b):
 """
 
 inputs = tokenizer.encode(input, return_tensors="pt").to(device)
-outputs = model.generate(inputs, max_length=128)
+outputs = model.generate(
+            inputs,
+            max_length=128,
+            num_beams=8,
+            early_stopping=True,
+            no_repeat_ngram_size=3,
+            pad_token_id=tokenizer.pad_token_id)
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 # Calculate the sum of two numbers.