Even though it appears that our little solution is working correctly, we still have to keep in mind that the model has been trained on a very small amount of data.
To improve the model, from the Default Custom Model page, you can click on the blue button labeled Edit and Retrain:
This will make our project editable.