Update main.py

main.py

@@ -33,7 +33,7 @@ async def prediction(image: UploadFile = File(...)):
     # process image
     image = Image.open(BytesIO(image))
     image = image.resize((image_shape[0], image_shape[1]))
-    image = np.expand_dims(image, axis=2)
+
 
     image = np.expand_dims(image, axis=0)
     prediction = model.predict(image)[0]