dpatel9923 commited on
Commit
ce5c85f
·
verified ·
1 Parent(s): 1f4d06f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -1
main.py CHANGED
@@ -33,7 +33,6 @@ async def prediction(image: UploadFile = File(...)):
33
  # process image
34
  image = Image.open(BytesIO(image))
35
  image = image.resize((image_shape[0], image_shape[1]))
36
- image = image.convert('L')
37
  image = np.expand_dims(image, axis=2)
38
 
39
  image = np.expand_dims(image, axis=0)
 
33
  # process image
34
  image = Image.open(BytesIO(image))
35
  image = image.resize((image_shape[0], image_shape[1]))
 
36
  image = np.expand_dims(image, axis=2)
37
 
38
  image = np.expand_dims(image, axis=0)