ehsanulhaque92/house-plant-api
0
๐ฟ House Plant Species Identifier API
This repository contains the containerized Flask backend for the House Plant Species Identifier, a full-stack, decoupled machine learning application.
Project Overview
This API serves a fine-tuned PyTorch (EfficientNet-B0) model capable of identifying 47 different species of house plants. It receives an image file via a POST request to the /predict endpoint and returns the predicted species and a confidence score in JSON format.
This backend is designed to be a standalone microservice, containerized with Docker, and is currently deployed on Hugging Face Spaces.
๐ Project Links
Technology Stack
- Backend: Python, Flask
- Machine Learning: PyTorch, Torchvision, Pillow
- Containerization: Docker
- Deployment: Hugging Face Spaces
