Update utils.py

utils.py

@@ -7,11 +7,13 @@ def preprocess_image(img_path):
     img = Image.open(img_path)
     img = img.resize((256, 256))
     img_array = np.array(img)
+    img_array = img_array[:, :, :3]  # Keep only the first three channels (RGB)
     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)