Docker wasn't working on my end, so I made it use a bash command for opening vision/start_gunicorn.sh.
I uploaded the python file shape_color_detection.py which runs the yolov8 model with the weights that I've uploaded in the file best.pt.
To test the model, I have uploaded two test images and changed test.py accordingly.
For requirements.txt, I prompt two more installs for torch and ultralytics to run the model.
Docker wasn't working on my end, so I made it use a bash command for opening vision/start_gunicorn.sh. I uploaded the python file shape_color_detection.py which runs the yolov8 model with the weights that I've uploaded in the file best.pt. To test the model, I have uploaded two test images and changed test.py accordingly. For requirements.txt, I prompt two more installs for torch and ultralytics to run the model.