Not getting the accuracy for Radioml_with_cnns on FPGA

[zsrabbani]

I've been using the RadioML CNN notebook as my reference to implement a model on an FPGA. I followed the standard structure to generate the bitstream and I've managed to get it running on the FPGA (AMD Zynq™ UltraScale+™ MPSoC ZCU104). The challenge I'm facing now is that the accuracy on the FPGA is only about 14%, even though the model itself achieves around 70% accuracy. So I'm looking for help in figuring out where I might be going wrong.

Here is my code: