### Ownership
This project is created by:
Samuel Nyberg: [email protected]
Tobias Mattsson: [email protected]

### install dependencies:

python version 3.12.8
python -m pip install -r requirements.txt

### To run notebooks. Follow steps:

1. Run data_prep.ipynb all cells, this will generate datasets to train the two models

2. Run regressor.ipynb
    - If you want to train the model yourself run from top to bottom.
    - If you wanna use pretrained model, skipp running cells that state "- Skip if using pretrained model"

3. Run classifier.ipynb
    - If you want to train the model yourself run from top to bottom.
    - If you wanna use pretrained model, skipp running cells that state "- Skip if using pretrained model"

4. Run combined_models.ipynb from top to bottom


To run flask_server_custom_model.ipynb above mentioned step 1-3 needs to be run in order, 
flask app will open local host webbapplication when following the link or open http://127.0.0.1:5000

To run yolo models run notebooks from top to bottom. If you want to retrain model, remove yolov8_custom_final.pt file