Update README.md

README.md

@@ -1,3 +1,19 @@
+---
+license: mit
+datasets:
+- xTRam1/safe-guard-prompt-injection
+language:
+- en
+metrics:
+- accuracy
+base_model:
+- FacebookAI/xlm-roberta-base
+pipeline_tag: text-classification
+library_name: keras
+tags:
+- cybersecurity
+- llmsecurity
+---
 # 🛡️ PromptShield
 
 **PromptShield** is a prompt classification model designed to detect **unsafe**, **adversarial**, or **prompt injection** inputs. Built on the `xlm-roberta-base` transformer, it delivers high-accuracy performance in distinguishing between **safe** and **unsafe** prompts — achieving **99.33% accuracy** during training.
@@ -105,5 +121,4 @@ print("🟢 Safe" if prediction == 0 else "🔴 Unsafe")
 
 📄 License
 
-MIT License
-
+MIT License