Hugging Face's logo

Spaces:
pushpinder06
/
Waste_classification
Sleeping

App Files Community
pushpinder06 commited on
Commit
1d2e0b6
·
verified ·
1 Parent(s): 5d7800c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  from PIL import Image
6
 
7
  # Load the trained model
8
- model = load_model("waste_classifier_model.h5")
9
  class_names = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
10
 
11
  # Prediction function
 
5
  from PIL import Image
6
 
7
  # Load the trained model
8
+ model = load_model("waste_classification(Mobilenetv2).h5")
9
  class_names = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
10
 
11
  # Prediction function