Update README.md

README.md

@@ -171,3 +171,86 @@ import torch
 tokenizer = AutoTokenizer.from_pretrained("Kelvinmbewe/LusakaLang-MultiTask")
 model = torch.load("Kelvinmbewe/LusakaLang-MultiTask/model.pt")
 model.eval()
+```
+
+```python
+predict_language([
+    "Ndeumfwa bwino lelo",
+    "Galimoto inachedwa koma driver anali bwino",
+    "The service was terrible today"
+])
+```
+
+
+```python
+predict_sentiment([
+    "Driver was rude and unprofessional",
+    "Ndimvela bwino lelo",
+    "The ride was okay, nothing special"
+])
+```
+
+```python
+predict_topic([
+    "Payment failed but money was deducted",
+    "Support siyankhapo, waited long",
+    "Driver was over speeding"
+])
+```
+
+
+```python
+@model{LusakaLangMultiTask,
+  author = {Kelvin Mbewe},
+  title = {LusakaLang Multi-Task Model},
+  year = 2025,
+  url = {https://huggingface.co/Kelvinmbewe/LusakaLang-MultiTask}
+}
+```
+
+
+```python
+
+                         ┌──────────────────────────────────────┐
+                         │         Input Text (Any Language)     │
+                         └──────────────────────────────────────┘
+                                         │
+                                         ▼
+                         ┌──────────────────────────────────────┐
+                         │        Tokenizer (mBERT-based)        │
+                         └──────────────────────────────────────┘
+                                         │
+                                         ▼
+                         ┌──────────────────────────────────────┐
+                         │      Shared mBERT Encoder Layer       │
+                         │  (bert-base-multilingual-cased)       │
+                         └──────────────────────────────────────┘
+                                         │
+                                         ▼
+                         ┌──────────────────────────────────────┐
+                         │        [CLS] Pooled Representation    │
+                         └──────────────────────────────────────┘
+                                         │
+         ┌───────────────────────────────┼───────────────────────────────┐
+         │                               │                               │
+         ▼                               ▼                               ▼
+┌──────────────────────┐     ┌────────────────────────┐     ┌────────────────────────┐
+│  Language Head        │     │  Sentiment Head        │     │   Topic Head            │
+│ (Kelvinmbewe/         │     │ (Kelvinmbewe/          │     │ (Kelvinmbewe/           │
+│  mbert_Lusaka_        │     │  mbert_LusakaLang_     │     │  mbert_LusakaLang_      │
+│  Language_Analysis)   │     │  Sentiment_Analysis)   │     │  Topic)                 │
+└──────────────────────┘     └────────────────────────┘     └────────────────────────┘
+         │                               │                               │
+         ▼                               ▼                               ▼
+┌──────────────────────┐     ┌────────────────────────┐     ┌────────────────────────┐
+│  Language Label       │     │  Sentiment Label       │     │  Topic Label            │
+│ (e.g., Bemba, Nyanja, │     │ (Negative/Neutral/     │     │ (Driver, Payment,       │
+│  English, Code‑Switch)│     │  Positive)             │     │  Support, etc.)         │
+└──────────────────────┘     └────────────────────────┘     └────────────────────────┘
+
+```
+
+
+
+
+