clamp probability to 0.0-1.0

app.py

@@ -43,7 +43,8 @@ def predict_image(img):
     img = img.convert("L")
     img = img.resize((28, 28))
     img = np.array(img)
-    return f"{learn.predict(img)[0][0]:.2f}"
+    pred = np.clip(learn.predict(img)[0][0], 0.0, 1.0)
+    return f"{pred:.2f}"
 
 @app.post("/predict")
 async def predict(data: ImageData):

model.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:04a212a1709f207d4f3c33a3ccd17d891df6917fe1ec1d7b0c1c918579b67df3
-size 89750534