diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py
index f9989cdc20df7aec0a6625b81ae4a94a0e39a671..f39a43ed3ecc4704d9c1f172ebe8c48f3f0e9c82 100644
--- a/modules/deeplearning/esrgan_exp.py
+++ b/modules/deeplearning/esrgan_exp.py
@@ -15,7 +15,8 @@ target_param = 'cld_opd_dcomp'
 # SAVED_MODEL_PATH = home_dir + '/esrgan-tf2_1'
 SAVED_MODEL_PATH = '/ships22/cloud/scratch/Satellite_Output/GOES-16/global/NREL_2023/2023_east_cf/tf_model_esrgan/esrgan'
 
-model = hub.load(SAVED_MODEL_PATH)
+# model = hub.load(SAVED_MODEL_PATH)
+model = tf.saved_model.load(SAVED_MODEL_PATH)
 
 
 def get_image_from_file(in_file):