readme: ---
  language: en
  ---
  ## Description
  This model is a ViT trained to classify waste images into 6 categories: 
  - Organic
  - Carton
  - Glass
  - General
  - Plastics
  - Dangerous.
  
  The repository related to this model is: https://github.com/hectorLop/Waste-Detector  
  Also, the code related to this model can be found here https://github.com/hectorLop/Waste-Detector/blob/main/waste_detector/classifier/sagemaker/model.py
  
  ### Requirements
  - Works with RGB images of size 224x224
  