Defkhan5960 commited on
Commit
8e9bd85
1 Parent(s): c7228b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from tensorflow.keras.models import load_model
8
  from tensorflow.keras.applications.efficientnet import preprocess_input
9
 
10
 
11
- model = load_model("efficent_net224B0.h5")
12
 
13
 
14
  waste_labels = {0: 'Fibres', 1: 'Nanowires', 2: 'Particles', 3: 'Powder'}
@@ -19,7 +19,7 @@ def classify_image(pil_image):
19
  img = image.img_to_array(pil_image)
20
 
21
 
22
- img = tf.image.resize(img, (224, 224))
23
 
24
 
25
  img = np.expand_dims(img, axis=0)
 
8
  from tensorflow.keras.applications.efficientnet import preprocess_input
9
 
10
 
11
+ model = load_model("efficent_netB7.h5")
12
 
13
 
14
  waste_labels = {0: 'Fibres', 1: 'Nanowires', 2: 'Particles', 3: 'Powder'}
 
19
  img = image.img_to_array(pil_image)
20
 
21
 
22
+ img = tf.image.resize(img, (600, 600))
23
 
24
 
25
  img = np.expand_dims(img, axis=0)