smart-sorting / README.md
karthikeya09's picture
Switch to Flask API with Docker
51087b5
metadata
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

POST /predict
{
    "image_base64": "<base64-encoded-image>"
}

Response

{
    "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
        }
    }
}