msank00/api_in_python

Expose python machine learning project as REST API

---

🚀 Simple Python API for ML project

• This meta project consists of simple script which can be used to create machine learning API in python which includes separate template for creating preprocessing, training and prediction API along with logging and basic error handling facility.
• The preprocessing, training, prediction module consists of dummy code which you can fill as per your wish.
• The APIs are created in 2 different ways:
  • Bare flask API
  • Swagger based API

⭐ Simple api using flask

• Kindly look into the simple_api/ directory here.

⭐ Swagger based api using FastAPI:

Swagger helps for API documentation which is quite helpful and a good practice.

• Kindly look into the swagger_api/ directory here.

⭐ Override FastAPI/Uvicorn logger using Loguru

• Please look here

⭐ Implement FastAPI with Redis (message queue approach)

• Please look here
• Multiple model syndrome and how to avoid them while deploying ML model using REST API ⚡
• Also includes load-testing using Locust ⚡

---

🔜 TODO

• Add requirement.txt file
• Implement with fastapi [click here]
  • Add swagger UI
• Add error code
• Add simple web UI
• Message Queue - Redis
• Load testing with Locust

---