Update README.md

README.md

@@ -20,7 +20,7 @@ A multi-task model to extract **causal attribution** from German texts.
 | **2. Relation Classification** | Sentence-Pair Classification | **14 Relation Labels** (e.g., MONO\_POS\_CAUSE, DIST\_NEG\_EFFECT, INTERDEPENDENCY, NO\_RELATION) |
 
 ## Usage
-Find the custom [library](https://github.com/padjohn/cbert). Once installed, run inference like so:
+Find the custom [library](https://github.com/padjohn/cbert/tree/v2.0). Once installed, run inference like so:
 ```python
 from causalbert.infer import load_model, sentence_analysis
 
@@ -52,4 +52,4 @@ print(all_results[0]['derived_relations'])
   - Batch size: 32
   - PEFT/LoRA: Enabled with r = 16
 
-See [train.py](https://github.com/padjohn/cbert/blob/main/causalbert/train.py) for the full configuration details.
+See [train.py](https://github.com/padjohn/cbert/blob/v2.0/causalbert/train.py) for the full configuration details.