Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ custom_objects = {'KerasLayer': hub.KerasLayer}
|
|
| 13 |
|
| 14 |
# Try loading the model with custom objects
|
| 15 |
try:
|
| 16 |
-
model = load_model('
|
| 17 |
except ValueError as e:
|
| 18 |
print("Model loading failed with error:", e)
|
| 19 |
print("Please ensure the model was saved correctly and matches the KerasLayer structure.")
|
|
|
|
| 13 |
|
| 14 |
# Try loading the model with custom objects
|
| 15 |
try:
|
| 16 |
+
model = load_model('bird_model.h5', custom_objects=custom_objects)
|
| 17 |
except ValueError as e:
|
| 18 |
print("Model loading failed with error:", e)
|
| 19 |
print("Please ensure the model was saved correctly and matches the KerasLayer structure.")
|