main py

app/main.py

@@ -329,7 +329,7 @@ async def handle_cnn(file: UploadFile = File(...)):
         # Convert predictions to label indexes
         predicted_label_indexes = np.argmax(predictions, axis=1)
         print(predicted_label_indexes)
-        
+        predicted_label_indexes = np.argmax(predicted_label_indexes)
         # Convert label indexes to actual label names
         predicted_labels = cnn_label_encoder.inverse_transform([predicted_label_indexes])