Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import numpy as np
|
|
5 |
import cv2
|
6 |
|
7 |
# Load the trained model
|
8 |
-
model_path = './
|
9 |
model = load_model(model_path)
|
10 |
|
11 |
class_names = ['ADONIS', 'AFRICAN GIANT SWALLOWTAIL', 'AMERICAN SNOOT', 'AN 88', 'APPOLLO', 'ARCIGERA FLOWER MOTH', 'ATALA', 'ATLAS MOTH', 'BANDED ORANGE HELICONIAN', 'BANDED PEACOCK']
|
|
|
5 |
import cv2
|
6 |
|
7 |
# Load the trained model
|
8 |
+
model_path = './model_checkpoint_manual_resnet.h5'
|
9 |
model = load_model(model_path)
|
10 |
|
11 |
class_names = ['ADONIS', 'AFRICAN GIANT SWALLOWTAIL', 'AMERICAN SNOOT', 'AN 88', 'APPOLLO', 'ARCIGERA FLOWER MOTH', 'ATALA', 'ATLAS MOTH', 'BANDED ORANGE HELICONIAN', 'BANDED PEACOCK']
|