Hugging Face's logo

Spaces:
Taidat
/
Image_Classification
Sleeping

App Files Community
Taidat commited on
Commit
18b1ad9
·
verified ·
1 Parent(s): e62ba89

Update app.py

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