Update README.md

README.md

@@ -1,3 +1,12 @@
+---
+license: mit
+library_name: peft
+tags:
+- text generation
+- nlp
+- llm
+- pii
+---
 # PII Guardian PHI3 Mini LORA
 
 This repository contains a fine-tuned model for detecting Personally Identifiable Information (PII) using PHI3 Mini with LoRA applied to the query, key, and value layers. The model is optimized for accuracy and efficiency in identifying various PII entities.
@@ -57,4 +66,4 @@ inputs = tokenizer(model_prompt, return_tensors="pt").to(device)
 # adjust max_new_tokens according to your need
 outputs = pii_model.generate(**inputs, do_sample=True, max_new_tokens=120)
 response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-print(response) #  {'firstname': ['Abner'], 'dob': ['23/03/1926'], 'email': ['[email protected]']}
+print(response) #  {'firstname': ['Abner'], 'dob': ['23/03/1926'], 'email': ['[email protected]']}