Fix Prediction.confidence should be float but not numpy.ndarray #7

Pandede · Sep 18, 2023 · 054edbf · 054edbf
1 parent 2553261
commit 054edbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wpodnet/backend.py b/wpodnet/backend.py
@@ -120,5 +120,5 @@ def predict(self, image: Image.Image) -> Prediction:
         return Prediction(
             image=image,
             bounds=bounds.astype(np.int32),
-            confidence=max_prob
+            confidence=max_prob.item()
         )