Update README.md

README.md

@@ -71,7 +71,7 @@ CALM-8B is trained on a **multi-task dataset** covering dialogue state tracking,
 
 ---
 ## Usage
-### 🏗 How to Load the Model
+### 🏗 How to Load the Model using Transformers
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
@@ -79,10 +79,20 @@ tokenizer = AutoTokenizer.from_pretrained("uiuc-convai/CALM-8B")
 model = AutoModelForCausalLM.from_pretrained("uiuc-convai/CALM-8B")
 ```
 
-<!-- TODO -->
-### 🛠 Example Inference
-```python
-TODO
+### 🛠 Example Oumi Inference
+```bash
+pip install oumi
+
+# See oumi_infer.yaml in this model's /oumi/ directory.
+oumi infer -i -c ./oumi_infer.yaml
+```
+
+### 🛠 Example Oumi Fine-Tuning
+```bash
+pip install oumi
+
+# See oumi_train.yaml in this model's /oumi/ directory.
+oumi train -c ./oumi_train.yaml
 ```
 
 ---