diff --git a/README.md b/README.md
index 2a67433..d8b8e0b 100644
--- a/README.md
+++ b/README.md
@@ -13,7 +13,7 @@ The Carvana data is available on the [Kaggle website](https://www.kaggle.com/c/c
 **Note : Use Python 3.6 or newer**
 ### Prediction
 
-You can easily test the output masks on your images via the CLI.
+After training your model and saving it to MODEL.pth, you can easily test the output masks on your images via the CLI.
 
 To predict a single image and save it: