fix: fix png file

main.py

@@ -15,14 +15,14 @@ with open("labels.txt") as f:
 def classify_image(img):
     # Resize the input image to the expected shape (224, 224)
     img_array = np.asarray(img.resize((224, 224)))[..., :3]
-    img_array = img_array.reshape((-1, 224, 224, 3))
+    img_array = img_array.reshape((1, 224, 224, 3))  # Add batch dimension
     img_array = tf.keras.applications.efficientnet.preprocess_input(img_array)
     prediction = model.predict(img_array).flatten()
     confidences = {labels[i]: float(prediction[i]) for i in range(90)}
     # Sort the confidences dictionary by value and get the top 3 items
-    top_3_confidences = dict(sorted(confidences.items(), key=lambda item: item[1], reverse=True)[:3])
+    # top_3_confidences = dict(sorted(confidences.items(), key=lambda item: item[1], reverse=True)[:3])
     
-    return top_3_confidences
+    return confidences
 
 @app.post("/predict")
 async def predict(file: bytes = File(...)):