attilaultzindur/garbage-classifier

Garbage Classification Model

A ResNet50 model fine-tuned on the Garbage Classification dataset.

Model Details

• Base Architecture: ResNet50
• Input Size: 224x224
• Classes: battery, biological, cardboard, clothes, glass, metal, paper, plastic, shoes, trash

Usage

from transformers import pipeline
from PIL import Image

# Create pipeline
classifier = pipeline("image-classification", model="YOUR_USERNAME/garbage-classification")

# Load image
image = Image.open("test_image.jpg")

# Make prediction
result = classifier(image)
print(f"Prediction: {result[0]['label']}, Confidence: {result[0]['score']:.2f}")