smart-sorting / README.md
karthikeya09's picture
Switch to Flask API with Docker
51087b5
---
title: Smart Sorting
emoji: 🗑️
colorFrom: green
colorTo: blue
sdk: docker
pinned: false
license: mit
---
# Smart Waste Classifier API
AI-powered waste classification using MobileNetV2 (97.83% accuracy).
## API Endpoints
- `GET /` - Status check
- `POST /predict` - Classify an image
### Example Request
```json
POST /predict
{
"image_base64": "<base64-encoded-image>"
}
```
### Response
```json
{
"success": true,
"prediction": {
"class": "plastic",
"confidence": 95.5,
"all_probabilities": {
"glass": 1.2,
"metal": 0.5,
"non-recyclable": 0.3,
"organic": 0.8,
"paper": 1.7,
"plastic": 95.5
}
}
}
```