augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-24 07:36:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update app.py

Browse Source
This commit is contained in:
augustin64 2022-04-20 20:57:04 +02:00
parent 54851be216
commit 29f694d04d
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/webserver/app.py
View File

@ -37,14 +37,15 @@ def recognize_mnist(image):
write_image_to_binary(image, ".cache/image.bin")
try:
return {"status": 200, "data":
json.loads(subprocess.check_output([
'out/main',
'recognize',
'--modele', '.cache/reseau.bin',
'--in', '.cache/image.bin',
'--out', 'json'
]))["0"]}
output = subprocess.check_output([
'out/main',
'recognize',
'--modele', '.cache/reseau.bin',
'--in', '.cache/image.bin',
'--out', 'json'
]).decode("utf-8")
json_data = json.loads(output.replace("nan", "0.0"))["0"]
return {"status": 200, "data": json_data}
except subprocess.CalledProcessError:
return {
"status": 500,