Aryan-EcoClim commited on
Commit
3ce36d5
·
1 Parent(s): dfa03a1

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +0 -1
utils.py CHANGED
@@ -7,7 +7,6 @@ def preprocess_image(img_path):
7
  img = Image.open(img_path)
8
  img = img.resize((256, 256))
9
  img_array = np.array(img)
10
- img_array = np.expand_dims(img_array, axis=0)
11
  return img_array
12
 
13
  # Function to classify the garbage
 
7
  img = Image.open(img_path)
8
  img = img.resize((256, 256))
9
  img_array = np.array(img)
 
10
  return img_array
11
 
12
  # Function to classify the garbage