Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ import cv2
|
|
11 |
custom_objects = {'KerasLayer': hub.KerasLayer}
|
12 |
|
13 |
# Load your model (ensure the path is correct)
|
14 |
-
model = load_model('
|
15 |
|
16 |
# Define your class labels or categories for predictions
|
17 |
train_info = [] # Replace with your actual class labels
|
|
|
11 |
custom_objects = {'KerasLayer': hub.KerasLayer}
|
12 |
|
13 |
# Load your model (ensure the path is correct)
|
14 |
+
model = load_model('bird_model4.h5', custom_objects=custom_objects)
|
15 |
|
16 |
# Define your class labels or categories for predictions
|
17 |
train_info = [] # Replace with your actual class labels
|