Overglitch commited on
Commit
e322f8c
·
verified ·
1 Parent(s): a0f8100

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -50,8 +50,9 @@ def get_fingerprint_type(winner):
50
  async def predict_fingerprint_api(file: UploadFile = File(...)):
51
  try:
52
  contents = await file.read()
53
- print(f"CONTENTS\n\n{contents}")
54
- image = Image.open(BytesIO(contents)).convert('L')
 
55
  print(f"IMAGEN\n\n{image}")
56
  fingerprint_type = predict_fingerprint(image)
57
  return {"prediction": fingerprint_type}
 
50
  async def predict_fingerprint_api(file: UploadFile = File(...)):
51
  try:
52
  contents = await file.read()
53
+ im = Image.fromarray(BytesIO(contents))
54
+ print(f"CONTENTS\n\n{im}")
55
+ image = Image.open(im).convert('L')
56
  print(f"IMAGEN\n\n{image}")
57
  fingerprint_type = predict_fingerprint(image)
58
  return {"prediction": fingerprint_type}