diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
index b304d56120b7fd9a1caa2683bf6eae083741b0af..68f3342a45d7c79fa4050f6a10855ba034d91658 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
@@ -740,7 +740,6 @@ class SRCNN:
         print(refl.shape)
         refl = refl[1912:3512, 1912:3512]
         LEN_Y, LEN_X = refl.shape
-        LEN_Y_in, LEN_X_in = LEN_Y, LEN_X
         print(refl.shape)
         bt = get_grid_values_all(h5f, 'temp_11_0um_nom')
         bt = bt[1912:3512, 1912:3512]
@@ -762,7 +761,7 @@ class SRCNN:
         LEN_X = 2 * (LEN_X - 8)
 
         t0 = time.time()
-        cld_opd_sres = self.run_inference_(bt, refl, refl_sub_lo, refl_sub_hi, refl_sub_std, cld_opd, LEN_Y, LEN_X)
+        cld_opd_sres, LEN_Y_in, LEN_X_in = self.run_inference_(bt, refl, refl_sub_lo, refl_sub_hi, refl_sub_std, cld_opd, LEN_Y, LEN_X)
         t1 = time.time()
         print('inference time: ', (t1 - t0))
         print(cld_opd_sres.shape)
@@ -841,7 +840,7 @@ class SRCNN:
         cld_opd_sres = self.do_inference(data)
         cld_opd_sres = denormalize(cld_opd_sres, label_param, mean_std_dct)
 
-        return cld_opd_sres
+        return cld_opd_sres, bt_us.shape[0,]
 
 
 def run_restore_static(directory, ckpt_dir, out_file=None):