add logging

main.py

@@ -113,6 +113,6 @@ async def predict(request: PredictRequest):
         probabilities = torch.nn.functional.softmax(outputs, dim=1)
         predicted_probabilities = probabilities.numpy().tolist()
         confidence = round(predicted_probabilities[0][1], 2)
-
+    logging.info("confidence: %s", confidence)
     # Return the probabilities as JSON
     return JSONResponse(content={"confidence": confidence})