diff --git a/server.py b/server.py index b2255c4..0c6900d 100644 --- a/server.py +++ b/server.py @@ -38,5 +38,5 @@ def store_img(image): async def predict(image:UploadFile): tensor_image, raw_image = process_image(image) prediction = mnist_classifier.predict(tensor_image) - store_img(raw_image) + # store_img(raw_image) return {"prediction": prediction} \ No newline at end of file