Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
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, (
|
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)
|