drive with model.pth: https://drive.google.com/file/d/1REXie3hhC-7s_b3Zv7Gggnv5s5XwkupF/view?usp=drive_link

git clone https://github.com/Sonnguyen4002/BKAI.git run final.ipynb python3 infer.py --image_path image.jpeg

Note: you have to get the model.pth to run infer.py, it is in the drive, put it in this folder