Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,6 +21,7 @@ model_path = r"inceptionv3_class.h5"
|
|
| 21 |
print(model_path)
|
| 22 |
hierarchical_models['class'] = load_model(model_path)
|
| 23 |
print(hierarchical_models)
|
|
|
|
| 24 |
|
| 25 |
def load_and_preprocess_image(image, target_size=(299, 299)):
|
| 26 |
img_array = img_to_array(image)
|
|
@@ -40,6 +41,7 @@ def predict(image):
|
|
| 40 |
print(results)
|
| 41 |
print(class_label)
|
| 42 |
# Predict species level
|
|
|
|
| 43 |
species_preds = hierarchical_models[class_label].predict(image_array)
|
| 44 |
species_idx = np.argmax(species_preds)
|
| 45 |
species_label = labels[class_label][species_idx]
|
|
|
|
| 21 |
print(model_path)
|
| 22 |
hierarchical_models['class'] = load_model(model_path)
|
| 23 |
print(hierarchical_models)
|
| 24 |
+
|
| 25 |
|
| 26 |
def load_and_preprocess_image(image, target_size=(299, 299)):
|
| 27 |
img_array = img_to_array(image)
|
|
|
|
| 41 |
print(results)
|
| 42 |
print(class_label)
|
| 43 |
# Predict species level
|
| 44 |
+
hierarchical_models[class_label] = load_model(f"inceptionv3_{class_label}.h5")
|
| 45 |
species_preds = hierarchical_models[class_label].predict(image_array)
|
| 46 |
species_idx = np.argmax(species_preds)
|
| 47 |
species_label = labels[class_label][species_idx]
|