Creating a new extension module for real-time denoising integrating trained model (.h5 or .pth Model) into a custom module

I am currently working on integrating a “real-time denoising” into “3D Slicer (v 5.4.0)” by creating a “new extension module”. I have a trained “TensorFlow model saved in ‘.h5’ format” that I would like to deploy within a custom module to enhance image quality as images are streamed into the software.

I followed the SegmentationUNet scripted module as a reference and modified the code to implement a denoising module using my trained model. However, I encountered several errors in the console, the output volume displays as ‘No Image,’ and the output volume data streaming stops.

If there are any related “GitHub repositories, or documentation references” for my issue, I would greatly appreciate it.