From 3bcac80a21470b336c9145937a1f0623fae0dcef Mon Sep 17 00:00:00 2001 From: carlfeynman Date: Sun, 17 Sep 2023 09:12:09 +0530 Subject: [PATCH] log --- server.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/server.py b/server.py index be8811d..cae86a4 100644 --- a/server.py +++ b/server.py @@ -36,9 +36,11 @@ def store_img(image): @app.post("/predict") async def predict(image: UploadFile): - tensor_image, raw_image = process_image(image) - print(tensor_image.shape) - prediction = mnist_classifier.predict(tensor_image) + print(image) + # tensor_image, raw_image = process_image(image) + # print(tensor_image.shape) + # prediction = mnist_classifier.predict(tensor_image) # store_img(raw_image) + prediction = [] return {"prediction": prediction}