pip install -r requirements.txt

Train: python main.py

Run the model: python predictor.py

The model predicts the image saved in file cap.png