Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ os.makedirs(app.config["UPLOAD_FOLDER"], exist_ok=True)
|
|
| 21 |
# -----------------------------
|
| 22 |
|
| 23 |
# Load Keras classification model
|
| 24 |
-
best_model = load_model("efficientnet_b0_best.
|
| 25 |
IMG_SIZE = 128
|
| 26 |
|
| 27 |
CLASS_LABELS = ['biological', 'brown-glass', 'cardboard', 'green-glass',
|
|
|
|
| 21 |
# -----------------------------
|
| 22 |
|
| 23 |
# Load Keras classification model
|
| 24 |
+
best_model = load_model("efficientnet_b0_best.h5")
|
| 25 |
IMG_SIZE = 128
|
| 26 |
|
| 27 |
CLASS_LABELS = ['biological', 'brown-glass', 'cardboard', 'green-glass',
|