fix: fix img reshape

main.py

@@ -14,8 +14,7 @@ with open("labels.txt") as f:
 
 def classify_image(img):
     # Resize the input image to the expected shape (224, 224)
-    img_resized = img.resize((224, 224))
-    img_array = np.array(img_resized)
+    img_array = np.asarray(img.resize((224, 224)))[..., :3]
     img_array = img_array.reshape((-1, 224, 224, 3))
     img_array = tf.keras.applications.efficientnet.preprocess_input(img_array)
     prediction = model.predict(img_array).flatten()
@@ -23,8 +22,7 @@ def classify_image(img):
     return confidences
 
 @app.post("/predict")
-async def predict(file: UploadFile = File(...)):
-    contents = await file.read()
-    img = Image.open(BytesIO(contents))
+async def predict(file: bytes = File(...)):
+    img = Image.open(BytesIO(file))
     confidences = classify_image(img)
     return confidences