fix sheikh name

app/main.py

@@ -311,12 +311,13 @@ async def handle_cnn(file: UploadFile = File(...)):
         predicted_labels = cnn_label_encoder.inverse_transform(predicted_label_indexes)
 
         print('decoded', predicted_labels)
-        print('arg', np.argmax(predicted_labels, axis=1))
+        print('arg', np.argmax(predicted_labels))
+        sheikh_name = np.argmax(predicted_labels)
         # .tolist()
         # Clean up the temporary file
         os.remove(temp_filename)
         # Return a successful response with decoded predictions
-        return {"message": "File processed successfully", "sheikh": predicted_labels}
+        return {"message": "File processed successfully", "sheikh": sheikh_name}
     except Exception as e:
         print(e)
         # Handle possible exceptions