added model loading function

enhance_me/mirnet/mirnet.py

@@ -5,7 +5,7 @@ from typing import List
 from datetime import datetime
 
 from tensorflow import keras
-from tensorflow.keras import optimizers
+from tensorflow.keras import optimizers, models
 
 from wandb.keras import WandbCallback
 
@@ -76,6 +76,16 @@ class MIRNet:
             metrics=[peak_signal_noise_ratio],
         )
 
+    def load_model(
+        self, filepath, custom_objects=None, compile=True, options=None
+    ) -> None:
+        self.model = models.load_model(
+            filepath=filepath,
+            custom_objects=custom_objects,
+            compile=compile,
+            options=options,
+        )
+
     def save_weights(self, filepath, overwrite=True, save_format=None, options=None):
         self.model.save_weights(
             filepath, overwrite=overwrite, save_format=save_format, options=options