Update README.md

README.md

@@ -49,7 +49,7 @@ def generate_prompt(text):
     inputs = tokenizer(text, return_tensors='pt')
     input_ids = inputs.input_ids.to(device)
     attention_mask = inputs.attention_mask.to(device)
-    output = model.generate(input_ids, attention_mask=attention_mask, max_length=2048, eos_token_id=tokenizer.eos_token_id)
+    output = model.generate(input_ids, attention_mask=attention_mask, repetition_penalty=1.05, max_length=2048, eos_token_id=tokenizer.eos_token_id)
 
     return tokenizer.decode(output[0], skip_special_tokens=False)