Skip to content

Commit

Permalink
log
Browse files Browse the repository at this point in the history
  • Loading branch information
arun477 committed Sep 17, 2023
1 parent d38f6d4 commit 3bcac80
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions server.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}

0 comments on commit 3bcac80

Please sign in to comment.