Update main.py
Browse files
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)
|