diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index 114f857..cd35428 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -314,7 +314,8 @@ def save_samples(filepath, save_list):
 
 def plot_predictions(filepath, input_items, colormap, model):
 	filepath_jsonl = filepath.replace("_$$", "").replace(".png", ".jsonl")
-	os.truncate(filepath_jsonl, 0)
+	if os.path.exists(filepath_jsonl):
+		os.truncate(filepath_jsonl, 0)
 	
 	i = 0
 	for input_pair in input_items: