Update README.md

README.md

@@ -94,7 +94,7 @@ 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.85)
+output_entities = model.generate(inputs.input_ids, max_new_tokens=250, do_sample=True, temperature=0.7)
 raw_output_entities_text = tokenizer.decode(output_entities[0])
 entities = extract_last_assistant_response(raw_output_entities_text)