Update README.md

README.md

@@ -187,13 +187,75 @@ This repository contains:
 ---
 
 # Usage
-
 Example:
+## Folder Structure
+
+```text
+HomayShield/
+│
+├── datasets/
+│   ├── token_level_adversarial_tr_v2.jsonl
+│   ├── token_level_adversarial_en_v2.jsonl
+│   └── final_classifier_merged_all.jsonl
+│
+├── output/
+│   └── Homayv6/
+│       ├── homayshield_encoder.pt
+│       ├── homayshield_classifier.pt
+│       └── homayshield_attack_bank.npy
+│
+├── training2.py
+├── inference.py
+└── requirements.txt
+```
+
+---
 
-```python
-python inference2.py
+## Training Command
+
+```bash
+python training2.py \
+  --train \
+  ./datasets/token_level_adversarial_tr_v2.jsonl \
+  ./datasets/token_level_adversarial_en_v2.jsonl \
+  ./datasets/final_classifier_merged_all.jsonl \
+  --output-dir ./output/Homayv6
 ```
 
+---
+
+## Output Files After Training
+
+Training generates:
+
+```text
+output/Homayv6/
+├── homayshield_encoder.pt
+├── homayshield_classifier.pt
+└── homayshield_attack_bank.npy
+```
+
+---
+
+## Inference Command
+
+```bash
+python inference.py
+```
+
+Inference loads:
+
+* `homayshield_encoder.pt`
+* `homayshield_classifier.pt`
+* `homayshield_attack_bank.npy`
+
+from:
+
+```text
+./output/Homayv6/
+```
+
+
 Inference modes:
 
 * OR