Distracted driver detection with Jetson Nano

Please review my pet project for distracted driver detection. I have used the Resnet model for it.
Code: https://github.com/gasatig/Jetson/blob/master/driver_resnet.ipynb
Pre-trained Model: resnet.h5 - Google Drive
Data Source: State Farm Distracted Driver Detection | Kaggle
Please check the last cell for accuracy and cross-entropy for the model.