I had same problem and fixed by making sure that tensorflow and keras versions are consistent. tensorflow==2.4.1 and keras==2.4.
https://stackoverflow.com/questions/73796400/error-no-module-named-tensorflow-keras