Update README.md

README.md

@@ -60,7 +60,8 @@ def generate_response(input_text: str) -> str:
         do_sample=True,
         top_k=1,
         temperature=0.5,
-        max_new_tokens=300, 
+        max_new_tokens=300,
+        repetition_penalty=1.1,
         pad_token_id=tokenizer.eos_token_id)
 
     # Tokenize input