This repository contains only our proposed model with the parameters hard coded in the source file. In near future, an automatic training code will be uploaded.
Run the following command to install dependencies:
pip install numpy matplotlib torch torchvision sentence-transformers efficientnet_pytorch