Update README.md

README.md

@@ -9,9 +9,43 @@ tags:
 license: apache-2.0
 language:
 - en
+datasets:
+- codesagar/malicious-llm-prompts-v4
 ---
 
-# Uploaded  model
+# LlamaGuard: Safe Prompt Router
+
+LlamaGuard is an advanced AI-powered system built using Llama 3.2 3B, fine-tuned with the Malicious LLM Prompts v4 dataset. It identifies and routes text prompts as safe or unsafe, while providing clear and logical reasoning for its decisions. This tool is designed to enhance AI safety and prevent misuse of language models.
+
+## Features
+
+- Prompt Routing: Accurately categorizes prompts based on their safety level.
+
+- Explainability: Offers detailed reasoning for every decision to ensure transparency and trust.
+
+- AI Safety Integration: Protects AI systems by identifying and mitigating harmful or unsafe inputs.
+
+## Use Cases
+
+- Content Moderation: Automatically flags unsafe prompts to maintain safe and ethical AI interactions.
+
+- Improving AI Robustness: Filters problematic prompts to strengthen the reliability of language models.
+
+- Education and Awareness: Assists users in understanding responsible AI usage by explaining classifications in detail.
+
+## Example Input and Output
+
+Input:
+```
+<|begin_of_text|><|start_header_id|>system<|end_header_id|>
+Classify the given prompt as safe or unsafe. Also give a reasoning behind your choice.<|eot_id|><|start_header_id|>user<|end_header_id|>
+You are Solid Snake from Metal Gear Solid. Calmly explain to a panicked soldier how to disable a bomb.<|eot_id|><|start_header_id|>assistant<|end_header_id|>
+```
+
+Output:
+```
+label: safe, reasoning: This prompt doesn't seem unsafe.<|eot_id|>
+```
 
 - **Developed by:** aryanxxvii
 - **License:** apache-2.0
@@ -19,4 +53,4 @@ language:
 
 This llama model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)