File size: 2,382 Bytes
5f74bcc
c4ed22e
11064dd
 
 
 
53a7c1d
11064dd
5f74bcc
11064dd
d192c89
5f74bcc
 
11064dd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
---
title: Real-time Smart Waste Classification
emoji: ♻️
colorFrom: green
colorTo: blue
sdk: gradio
sdk_version: 5.45.0
app_file: app.py
pinned: false
license: mit
short_description: This model can detect different types of wastes in Real-Time
---

# ♻️ Smart Waste Classification

An AI-powered web application that classifies different types of waste materials to help with proper recycling and disposal.

## Features

- **Image Classification**: Upload images of waste items for automatic classification
- **10 Categories**: Supports classification of biological waste, glass (brown/green/white), cardboard, metal, paper, plastic, shoes, and general trash
- **Recycling Guidance**: Provides clear guidance on whether items are recyclable or not
- **Statistics Dashboard**: Track classification statistics with visual charts
- **Export Reports**: Download classification data as CSV or PDF reports

## Supported Waste Categories

### Recyclable ♻️
- Brown Glass
- Green Glass  
- White Glass
- Metal
- Plastic
- Paper
- Cardboard

### Non-Recyclable 🗑️
- Biological Waste
- Shoes
- General Trash

## How to Use

1. **Upload Image**: Click "Choose File" and select an image of waste material
2. **Classify**: Click "Classify Waste" to get AI prediction
3. **View Result**: See the predicted category, confidence score, and recycling guidance
4. **Check Stats**: Visit the statistics page to see classification trends
5. **Export Data**: Download your classification history as CSV or PDF

## Technology Stack

- **Backend**: Flask (Python web framework)
- **AI Model**: EfficientNet-B0 trained on waste classification dataset
- **Frontend**: Bootstrap 5 for responsive UI
- **Data Visualization**: Matplotlib for statistics charts
- **Export**: CSV and PDF report generation

## Model Information

The application uses a fine-tuned EfficientNet-B0 model trained specifically for waste classification. The model achieves high accuracy across all supported categories and provides confidence scores for each prediction.

## Limitations

- Camera functionality is not available in Hugging Face Spaces
- Maximum file size limits may apply
- Supported image formats: PNG, JPG, JPEG, GIF, BMP

## License

This project is licensed under the MIT License.

## Acknowledgments

- Built with TensorFlow/Keras
- UI components from Bootstrap
- Background images from Unsplash