diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index cd35428..9973c87 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -11,6 +11,7 @@ from lib.ai.components.CallbackCustomModelCheckpoint import CallbackCustomModelC
 import os
 import io
 import math
+import json
 import cv2
 import numpy as np
 from glob import glob