diff --git a/README.md b/README.md
index 4666b94..201e2e7 100644
--- a/README.md
+++ b/README.md
@@ -156,8 +156,8 @@ python hf-inference-example.py
 
 ### Training Hugging Face 🤗 version
 
-Prepare your dataset, edit the training example to set your dataset file and launch:
-
+Prepare your dataset, edit the training example to set your dataset file and launch training.
+![image](https://user-images.githubusercontent.com/22396871/226167997-475b806a-e257-4628-979c-d15df4b3bc5c.png)
 ```
 python hf-training-example.py
 ```