Deploy Classification Neural Network on Serverless GPU platform of Banana De

Setup

1. Download the repository

$ git clone <repo name>

2. After downloading navigate to te container ocr folder and run the dollowing commands

$ python -m venv mtailor
$ source mtailor/bin/activate 

3. Download Model weights into models/ directory

https://www.dropbox.com/s/b7641ryzmkceoc9/pytorch_model_weights.pth?dl=0

Development Environment Setup

4. Install required dependencies

pip install -r requirements.txt

5. Convert Pytorch model to Onnx

python src/utils/convert_to_onnx.py

6. Running tests

pytest src/tests/