fix decoding

app/main.py

@@ -329,9 +329,16 @@ async def handle_cnn(file: UploadFile = File(...)):
         # Convert predictions to label indexes
         predicted_label_indexes = np.argmax(predictions, axis=1)
         print(predicted_label_indexes)
+        unique_labels, counts = np.unique(predicted_label_indexes, return_counts=True)
+
+        # Step 2: Find the index of the maximum count
+        index_of_max_freq = np.argmax(counts)
+        # Step 3: Retrieve the most frequent item (index)
+        most_frequent_label_index = unique_labels[index_of_max_freq]
+        
         # 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)
+        predicted_labels = cnn_label_encoder.inverse_transform([most_frequent_label_index])
 
         print('decoded', predicted_labels)
         reciter_name = predicted_labels[0]