Update app/src/predict_one_image_with_metrics.py

app/src/predict_one_image_with_metrics.py

@@ -628,9 +628,9 @@ def predict_with_model(model, X: Any) -> Tuple[str, float]:
     if hasattr(model, "predict_proba"):
         prob_good = float(model.predict_proba(X)[0, 1])
     else:
-        prob_good = float("nan")
+        prob_good = None
     
-    label = "good" if prob_good >= 0.5 else "bad"
+    label = "good" if (prob_good is not None and prob_good >= 0.5) else "bad"
     return label, prob_good