Update README.md

README.md

@@ -95,7 +95,8 @@ print(text_to_anonymize)
 # Step 1: Extracting entities from text
 prompt = f'USER: Resample the entities: {text_to_anonymize}\n\nASSISTANT:'
 inputs = tokenizer(prompt, return_tensors='pt').to('cuda:0')
-output_entities = model.generate(inputs.input_ids, max_new_tokens=250, do_sample=True, temperature=0.7)
+output_entities = model.generate(inputs.input_ids, max_new_tokens=300, do_sample=False, temperature=0.8, penalty_alpha=1.3, top_k=180, num_beams=5, repetition_penalty=2.3)
+
 raw_output_entities_text = tokenizer.decode(output_entities[0])
 entities = extract_last_assistant_response(raw_output_entities_text)