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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -35,8 +35,7 @@ def predict_fingerprint(image):
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)
41
  return processed_image_resized
42
 
@@ -50,10 +49,9 @@ async def predict_fingerprint_api(file: UploadFile = File(...)):
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:
 
35
 
36
  def preprocess_image(image):
37
  # Guardar la imagen en formato TIFF
38
+ processed_image = representativo("temp_image.tif")
 
39
  processed_image_resized = processed_image.reshape(1, -1)
40
  return processed_image_resized
41
 
 
49
  try:
50
  contents = await file.read()
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}
57
  except Exception as e: