Overglitch commited on
Commit
5f075c7
·
verified ·
1 Parent(s): 024a209

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,6 +69,7 @@ async def predict(file: UploadFile = File(...)):
69
  contents = await file.read()
70
  image = Image.open(BytesIO(contents)).convert('L')
71
  image = np.asarray(image)
 
72
  if image.shape != (256, 256):
73
  raise ValueError("La imagen debe ser de tamaño 256x256.")
74
  image = image.reshape(256, 256, 1)
 
69
  contents = await file.read()
70
  image = Image.open(BytesIO(contents)).convert('L')
71
  image = np.asarray(image)
72
+ image = cv2.resize(image, (256, 256), interpolation=cv2.INTER_AREA)
73
  if image.shape != (256, 256):
74
  raise ValueError("La imagen debe ser de tamaño 256x256.")
75
  image = image.reshape(256, 256, 1)