Update README.md

README.md

@@ -62,24 +62,6 @@ import opensportslib
 print("OpenSportsLib imported successfully")
 ```
 
-### Train a Localization model
-
-```python
-from opensportslib.apis import LocalizationModel
-
-my_model = LocalizationModel(
-    config="/path/to/localization.yaml",
-👉  weights="OpenSportsLab/OSL-loc-snbas-2023-e2e",  # optional
-)
-
-best_checkpoint = my_model.train(
-    train_set="/path/to/train.json",
-    valid_set="/path/to/valid.json",
-)
-
-print(best_checkpoint)
-```
-
 ### Run inference
 
 ```python
@@ -87,7 +69,7 @@ from opensportslib.apis import LocalizationModel
 
 my_model = LocalizationModel(
     config="/path/to/localization.yaml",
-    weights="OpenSportsLab/OSL-loc-snbas-2023-e2e",
+👉   weights="OpenSportsLab/OSL-loc-snbas-2023-e2e",
 )
 
 predictions = my_model.infer(