VishnuEcoClim commited on
Commit
6750728
·
1 Parent(s): 557f43d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from utils import * # Assuming the gen_labels() and preprocess() functions are
7
 
8
  # Function to classify the garbage
9
  def classify_garbage(img_path, model):
10
- processed_img = preprocess_image(img_path)
11
  prediction = model.predict(processed_img)
12
 
13
  class_labels = ["cardboard", "glass", "metal", "paper", "plastic", "trash"]
 
7
 
8
  # Function to classify the garbage
9
  def classify_garbage(img_path, model):
10
+ processed_img = preprocess(img_path)
11
  prediction = model.predict(processed_img)
12
 
13
  class_labels = ["cardboard", "glass", "metal", "paper", "plastic", "trash"]