Overglitch commited on
Commit
dcb2259
·
verified ·
1 Parent(s): fceb036

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -51,6 +51,7 @@ async def predict_fingerprint_api(file: UploadFile = File(...)):
51
  image = Image.open(BytesIO(contents)).convert('L')
52
  image.resize((256,256))
53
  image.save("temp_image.tif")
 
54
  print(f"\n\n\n\nSIZE:{image.size}")
55
  fingerprint_type = predict_fingerprint(image)
56
  return {"prediction": fingerprint_type}
 
51
  image = Image.open(BytesIO(contents)).convert('L')
52
  image.resize((256,256))
53
  image.save("temp_image.tif")
54
+ image = Image.open("temp_image.tiff")
55
  print(f"\n\n\n\nSIZE:{image.size}")
56
  fingerprint_type = predict_fingerprint(image)
57
  return {"prediction": fingerprint_type}