This repository contains a pretrained ResNet18 model for detecting white rice in a give image.
- src/TrainAndEvaluate.ipynb has the details of model training and evaluation
- In order to test the model use src/server_model.py by running
cd src streamlit run serve_model.py