diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index 294cab4..c247154 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -262,7 +262,10 @@ def plot_predictions(filepath, input_items, colormap, model):
 		
 		plot_samples_matplotlib(
 			filepath,
-			[input_tensor, prediction_colormap],
+			[
+				# input_tensor,
+				prediction_colormap
+			],
 			figsize=(18, 14)
 		)