Building API-endpoint for image classification, when running main.py it checks for the model in onnx format in your directory if it's not there already it will downlaod it from torch model zoo and change it into onnx format and then the API is in ACTION !!!
preview.mp4
-
pip install -r requirements.txt -
Run
main.pythis will run uvicorn server with the endpoints. -
Open browser and navigate to the follwoing localhost for swagger ui
http://127.0.0.1:8000/docs.