Hugging Face's logo

Spaces:
Taidat
/
Image_Classification
Sleeping

App Files Community
Taidat commited on
Commit
725e77d
·
verified ·
1 Parent(s): 0a9ed4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  from PIL import Image
5
 
6
  # Load model
7
- model = tf.keras.models.load_model("model.h5")
8
 
9
  # Classes in training order
10
  class_names = ['batteries', 'clothes', 'e-waste', 'glass', 'light blubs', 'metal', 'organic', 'paper', 'plastic']
 
4
  from PIL import Image
5
 
6
  # Load model
7
+ model = tf.keras.models.load_model("model_compatible.h5")
8
 
9
  # Classes in training order
10
  class_names = ['batteries', 'clothes', 'e-waste', 'glass', 'light blubs', 'metal', 'organic', 'paper', 'plastic']