Update README.md

README.md

@@ -5,4 +5,8 @@ language: en
 This model is a ViT trained to detect waste from images.
 
 The repository related to this model is: https://github.com/hectorLop/Waste-Detector  
-This model was created using [Icevision](https://github.com/airctic/icevision), and all the code related to the training can be found here https://github.com/hectorLop/Waste-Detector/tree/main/waste_detector/object_detection/sagemaker
+This model was created using [Icevision](https://github.com/airctic/icevision), and all the code related to the training can be found here https://github.com/hectorLop/Waste-Detector/tree/main/waste_detector/object_detection/sagemaker
+
+### Requirements
+- It is an EffientDet D1 model 
+- Works with RBG images of size 512x512