Missing "python" in code segments

app/src/content/chapters/sft.mdx

@@ -436,7 +436,7 @@ training_args = SFTConfig(
 
 Then we can train the model like this and TRL will deal with data loading, batching, and training.
 
-```
+```python
 from trl import SFTTrainer
 
 trainer = SFTTrainer(
@@ -449,6 +449,6 @@ trainer = SFTTrainer(
 
 And then we can train the model like this:
 
-```
+```python
 trainer_stats = trainer.train()
 ```