Update utils.py

utils.py

@@ -7,8 +7,11 @@ def preprocess_image(img_path):
     img = Image.open(img_path)
     img = img.resize((256, 256))
     img_array = np.array(img)
+    img_array = np.expand_dims(img_array, axis=0)  # Add an extra dimension for batch size
+    img_array = img_array / 255.0  # Normalize pixel values to the range [0, 1]
     return img_array
 
+
 # Function to classify the garbage
 def classify_garbage(img_path, model):
     processed_img = preprocess_image(img_path)