Overglitch commited on
Commit
0ed0acb
·
verified ·
1 Parent(s): cbb5314

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -35,7 +35,6 @@ def predict_fingerprint(image):
35
 
36
  def preprocess_image(image):
37
  # Guardar la imagen en formato TIFF
38
- image.resize((256,256))
39
  image.save("temp_image.tiff")
40
  processed_image = representativo("temp_image.tiff")
41
  processed_image_resized = processed_image.reshape(1, -1)
@@ -51,7 +50,10 @@ async def predict_fingerprint_api(file: UploadFile = File(...)):
51
  try:
52
  contents = await file.read()
53
  image = Image.open(BytesIO(contents)).convert('L')
54
- print(f"IMAGEN\n\n{list(image.getdata())}")
 
 
 
55
  fingerprint_type = predict_fingerprint(image)
56
  return {"prediction": fingerprint_type}
57
  except Exception as e:
 
35
 
36
  def preprocess_image(image):
37
  # Guardar la imagen en formato TIFF
 
38
  image.save("temp_image.tiff")
39
  processed_image = representativo("temp_image.tiff")
40
  processed_image_resized = processed_image.reshape(1, -1)
 
50
  try:
51
  contents = await file.read()
52
  image = Image.open(BytesIO(contents)).convert('L')
53
+ print(f"IMAGEN\n{image.size}\n{list(image.getdata())}")
54
+ Image.fromarray(list(image.getdata()))
55
+ image.resize((256,256))
56
+ print(f"NSIZE:{image.size}")
57
  fingerprint_type = predict_fingerprint(image)
58
  return {"prediction": fingerprint_type}
59
  except Exception as e: