Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def f1(y_true, y_pred):
|
|
28 |
# ---------------------------
|
29 |
# Load model with custom objects
|
30 |
# ---------------------------
|
31 |
-
model = load_model("
|
32 |
|
33 |
# ---------------------------
|
34 |
# Prediction function
|
|
|
28 |
# ---------------------------
|
29 |
# Load model with custom objects
|
30 |
# ---------------------------
|
31 |
+
model = load_model("pneumonia_cnn_model.h5", custom_objects={'f1': f1, 'precision': precision, 'recall': recall})
|
32 |
|
33 |
# ---------------------------
|
34 |
# Prediction function
|