diff --git a/aimodel/src/subcommands/rainfall_stats.py b/aimodel/src/subcommands/rainfall_stats.py
index 8bd8d1d..f128596 100755
--- a/aimodel/src/subcommands/rainfall_stats.py
+++ b/aimodel/src/subcommands/rainfall_stats.py
@@ -9,12 +9,7 @@ from loguru import logger
 import tensorflow as tf
 from lib.dataset.batched_iterator import batched_iterator
 
-from lib.io.handle_open import handle_open
-from lib.ai.RainfallWaterMono import RainfallWaterMono
 from lib.dataset.dataset_mono import dataset_mono_predict
-from lib.io.find_paramsjson import find_paramsjson
-from lib.io.readfile import readfile
-from lib.vis.segmentation_plot import segmentation_plot
 
 
 MODE_JSONL = 1