diff --git a/classifiers.py b/classifiers.py
index 81aa98b..b67f689 100644
--- a/classifiers.py
+++ b/classifiers.py
@@ -56,6 +56,7 @@ class YOLOClassify:
 
     def classify(self, yolo_classifier_model, image_to_classify):
         image_to_classify = image_to_classify.permute(0, 3, 1, 2)
+        image_to_classify = image_to_classify / 255
         image_to_classify = image_to_classify.squeeze(0)
         transform = transforms.Compose([
             transforms.Resize(size=(224, 224))