diff --git a/docs/README.md b/docs/README.md
index c2897d8..29dbe1a 100755
--- a/docs/README.md
+++ b/docs/README.md
@@ -40,25 +40,25 @@ unzip test_data.zip
 
 - For mseed format:
 ```
-python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/mseed.csv --data_dir=test_data/mseed --format=mseed --amplitude --response_xml=test_data/stations.xml --batch_size=1 --sampling_rate=100 --plot_figure
+python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/mseed.csv --data_dir=test_data/mseed --format=mseed --amplitude --response_xml=test_data/stations.xml --batch_size=1 --sampling_rate=100
 ```
 ```
-python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/mseed2.csv --data_dir=test_data/mseed --format=mseed --amplitude --response_xml=test_data/stations.xml --batch_size=1 --sampling_rate=100 --plot_figure
+python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/mseed2.csv --data_dir=test_data/mseed --format=mseed --amplitude --response_xml=test_data/stations.xml --batch_size=1 --sampling_rate=100
 ```
 
 - For sac format:
 ```
-python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/sac.csv --data_dir=test_data/sac --format=sac --batch_size=1 --plot_figure
+python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/sac.csv --data_dir=test_data/sac --format=sac --batch_size=1
 ```
 
 - For numpy format:
 ```
-python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/npz.csv --data_dir=test_data/npz --format=numpy --plot_figure
+python phasenet/predict.py --model=model/190703-214543 --data_list=test_data/npz.csv --data_dir=test_data/npz --format=numpy
 ```
 
 - For hdf5 format:
 ```
-python phasenet/predict.py --model=model/190703-214543 --hdf5_file=test_data/data.h5 --hdf5_group=data --format=hdf5 --plot_figure
+python phasenet/predict.py --model=model/190703-214543 --hdf5_file=test_data/data.h5 --hdf5_group=data --format=hdf5
 ```
 
 - For a seismic array (used by [QuakeFlow](https://github.com/wayneweiqiang/QuakeFlow)):
diff --git a/env.yml b/env.yml
index e29273a..b36d565 100644
--- a/env.yml
+++ b/env.yml
@@ -1,17 +1,193 @@
 name: phasenet
 channels:
-  - defaults
   - conda-forge
+  - defaults
 dependencies:
-  - python
-  - numpy
-  - scipy
-  - matplotlib
-  - pandas
-  - scikit-learn
-  - tqdm
-  - obspy
-  - uvicorn
-  - fastapi
-  - tensorflow
-  - keras
+  - _libgcc_mutex=0.1=conda_forge
+  - _openmp_mutex=4.5=2_gnu
+  - _tflow_select=2.3.0=mkl
+  - abseil-cpp=20211102.0=hd4dd3e8_0
+  - absl-py=2.1.0=py310h06a4308_0
+  - aiohappyeyeballs=2.4.0=py310h06a4308_0
+  - aiohttp=3.10.5=py310h5eee18b_0
+  - aiosignal=1.2.0=pyhd3eb1b0_0
+  - annotated-types=0.6.0=py310h06a4308_0
+  - anyio=4.6.2=py310h06a4308_0
+  - astunparse=1.6.3=py_0
+  - async-timeout=4.0.3=py310h06a4308_0
+  - attrs=24.2.0=py310h06a4308_0
+  - blas=1.0=mkl
+  - blinker=1.6.2=py310h06a4308_0
+  - bottleneck=1.4.2=py310ha9d4c09_0
+  - brotli=1.0.9=h5eee18b_8
+  - brotli-bin=1.0.9=h5eee18b_8
+  - brotli-python=1.0.9=py310h6a678d5_8
+  - bzip2=1.0.8=h5eee18b_6
+  - c-ares=1.19.1=h5eee18b_0
+  - ca-certificates=2024.9.24=h06a4308_0
+  - cachetools=5.3.3=py310h06a4308_0
+  - certifi=2024.8.30=py310h06a4308_0
+  - cffi=1.17.1=py310h1fdaa30_0
+  - charset-normalizer=3.3.2=pyhd3eb1b0_0
+  - click=8.1.7=py310h06a4308_0
+  - contourpy=1.3.1=py310hdb19cb5_0
+  - cryptography=41.0.3=py310h130f0dd_0
+  - cycler=0.11.0=pyhd3eb1b0_0
+  - cyrus-sasl=2.1.28=h9c0eb46_1
+  - dbus=1.13.18=hb2f20db_0
+  - decorator=5.1.1=pyhd3eb1b0_0
+  - exceptiongroup=1.2.0=py310h06a4308_0
+  - expat=2.6.3=h6a678d5_0
+  - fastapi=0.112.2=py310h06a4308_0
+  - flatbuffers=24.3.25=h6a678d5_0
+  - fontconfig=2.14.1=h4c34cd2_2
+  - fonttools=4.51.0=py310h5eee18b_0
+  - freetype=2.12.1=h4a9f257_0
+  - frozenlist=1.5.0=py310h5eee18b_0
+  - gast=0.4.0=pyhd3eb1b0_0
+  - giflib=5.2.2=h5eee18b_0
+  - glib=2.78.4=h6a678d5_0
+  - glib-tools=2.78.4=h6a678d5_0
+  - google-auth=2.29.0=py310h06a4308_0
+  - google-auth-oauthlib=0.4.4=pyhd3eb1b0_0
+  - google-pasta=0.2.0=pyhd3eb1b0_0
+  - greenlet=3.0.1=py310h6a678d5_0
+  - grpc-cpp=1.46.1=h33aed49_1
+  - grpcio=1.42.0=py310hce63b2e_0
+  - gst-plugins-base=1.14.1=h6a678d5_1
+  - gstreamer=1.14.1=h5eee18b_1
+  - h11=0.14.0=py310h06a4308_0
+  - h5py=3.12.1=py310hc0802c4_0
+  - hdf5=1.12.1=h70be1eb_2
+  - icu=58.2=he6710b0_3
+  - idna=3.7=py310h06a4308_0
+  - intel-openmp=2023.1.0=hdb19cb5_46306
+  - joblib=1.4.2=py310h06a4308_0
+  - jpeg=9e=h5eee18b_3
+  - keras=2.11.0=py310h06a4308_0
+  - keras-preprocessing=1.1.2=pyhd3eb1b0_0
+  - kiwisolver=1.4.4=py310h6a678d5_0
+  - krb5=1.20.1=h568e23c_1
+  - lcms2=2.12=h3be6417_0
+  - ld_impl_linux-64=2.40=h12ee557_0
+  - lerc=3.0=h295c915_0
+  - libbrotlicommon=1.0.9=h5eee18b_8
+  - libbrotlidec=1.0.9=h5eee18b_8
+  - libbrotlienc=1.0.9=h5eee18b_8
+  - libclang=14.0.6=default_hc6dbbc7_1
+  - libclang13=14.0.6=default_he11475f_1
+  - libcups=2.4.2=ha637b67_0
+  - libcurl=8.2.1=h91b91d3_0
+  - libdeflate=1.17=h5eee18b_1
+  - libedit=3.1.20230828=h5eee18b_0
+  - libev=4.33=h7f8727e_1
+  - libffi=3.4.4=h6a678d5_1
+  - libgcc=14.2.0=h77fa898_1
+  - libgcc-ng=14.2.0=h69a702a_1
+  - libgfortran-ng=11.2.0=h00389a5_1
+  - libgfortran5=11.2.0=h1234567_1
+  - libglib=2.78.4=hdc74915_0
+  - libgomp=14.2.0=h77fa898_1
+  - libiconv=1.16=h5eee18b_3
+  - libllvm14=14.0.6=hecde1de_4
+  - libnghttp2=1.52.0=ha637b67_1
+  - libpng=1.6.39=h5eee18b_0
+  - libpq=12.15=h37d81fd_1
+  - libprotobuf=3.20.3=he621ea3_0
+  - libssh2=1.10.0=h37d81fd_2
+  - libstdcxx-ng=11.2.0=h1234567_1
+  - libtiff=4.5.1=h6a678d5_0
+  - libuuid=1.41.5=h5eee18b_0
+  - libwebp-base=1.3.2=h5eee18b_1
+  - libxcb=1.15=h7f8727e_0
+  - libxkbcommon=1.0.1=h5eee18b_1
+  - libxml2=2.10.4=hcbfbd50_0
+  - libxslt=1.1.37=h2085143_0
+  - lxml=5.2.1=py310hdbbb534_0
+  - lz4-c=1.9.4=h6a678d5_1
+  - markdown=3.4.1=py310h06a4308_0
+  - markupsafe=2.1.3=py310h5eee18b_0
+  - matplotlib=3.8.4=py310h06a4308_0
+  - matplotlib-base=3.8.4=py310h1128e8f_0
+  - mkl=2023.1.0=h213fc3f_46344
+  - mkl-service=2.4.0=py310h5eee18b_1
+  - mkl_fft=1.3.11=py310h5eee18b_0
+  - mkl_random=1.2.8=py310h1128e8f_0
+  - multidict=6.1.0=py310h5eee18b_0
+  - mysql=5.7.24=he378463_2
+  - ncurses=6.4=h6a678d5_0
+  - numexpr=2.10.1=py310h3c60e43_0
+  - numpy=1.26.4=py310h5f9d8c6_0
+  - numpy-base=1.26.4=py310hb5e798b_0
+  - oauthlib=3.2.2=py310h06a4308_0
+  - obspy=1.4.1=py310h5b4e0ec_3
+  - openjpeg=2.5.2=he7f1fd0_0
+  - openssl=1.1.1w=h7f8727e_0
+  - opt_einsum=3.3.0=pyhd3eb1b0_1
+  - packaging=24.1=py310h06a4308_0
+  - pandas=2.1.4=py310h1128e8f_0
+  - pcre2=10.42=hebb0a14_1
+  - pillow=11.0.0=py310hfdbf927_0
+  - pip=24.2=py310h06a4308_0
+  - ply=3.11=py310h06a4308_0
+  - propcache=0.2.0=py310h5eee18b_0
+  - protobuf=3.20.3=py310h6a678d5_0
+  - pyasn1=0.4.8=pyhd3eb1b0_0
+  - pyasn1-modules=0.2.8=py_0
+  - pycparser=2.21=pyhd3eb1b0_0
+  - pydantic=2.8.2=py310h06a4308_0
+  - pydantic-core=2.20.1=py310hb02cf49_0
+  - pyjwt=2.9.0=py310h06a4308_0
+  - pyopenssl=23.2.0=py310h06a4308_0
+  - pyparsing=3.0.9=py310h06a4308_0
+  - pyqt=5.15.10=py310h6a678d5_0
+  - pyqt5-sip=12.13.0=py310h5eee18b_0
+  - pysocks=1.7.1=py310h06a4308_0
+  - python=3.10.13=h7a1cb2a_0
+  - python-dateutil=2.9.0post0=py310h06a4308_2
+  - python-flatbuffers=24.3.25=py310h06a4308_0
+  - python-tzdata=2023.3=pyhd3eb1b0_0
+  - python_abi=3.10=2_cp310
+  - pytz=2024.1=py310h06a4308_0
+  - qt-main=5.15.2=h5b8104b_9
+  - re2=2022.04.01=h295c915_0
+  - readline=8.2=h5eee18b_0
+  - requests=2.32.3=py310h06a4308_1
+  - requests-oauthlib=2.0.0=py310h06a4308_0
+  - rsa=4.7.2=pyhd3eb1b0_1
+  - scikit-learn=1.5.1=py310h1128e8f_0
+  - scipy=1.14.1=py310h5f9d8c6_0
+  - setuptools=75.1.0=py310h06a4308_0
+  - sip=6.7.12=py310h6a678d5_0
+  - six=1.16.0=pyhd3eb1b0_1
+  - snappy=1.2.1=h6a678d5_0
+  - sniffio=1.3.0=py310h06a4308_0
+  - sqlalchemy=1.4.51=py310h5eee18b_0
+  - sqlite=3.45.3=h5eee18b_0
+  - starlette=0.38.2=py310h06a4308_0
+  - tbb=2021.8.0=hdb19cb5_0
+  - tensorboard=2.11.0=py310h06a4308_0
+  - tensorboard-data-server=0.6.1=py310h52d8a92_0
+  - tensorboard-plugin-wit=1.8.1=py310h06a4308_0
+  - tensorflow=2.11.0=mkl_py310hb40ee82_0
+  - tensorflow-base=2.11.0=mkl_py310he5f8e37_0
+  - tensorflow-estimator=2.11.0=py310h06a4308_0
+  - termcolor=2.1.0=py310h06a4308_0
+  - threadpoolctl=3.5.0=py310h2f386ee_0
+  - tk=8.6.14=h39e8969_0
+  - tomli=2.0.1=py310h06a4308_0
+  - tornado=6.4.1=py310h5eee18b_0
+  - tqdm=4.66.5=py310h2f386ee_0
+  - typing-extensions=4.11.0=py310h06a4308_0
+  - typing_extensions=4.11.0=py310h06a4308_0
+  - tzdata=2024b=h04d1e81_0
+  - unicodedata2=15.1.0=py310h5eee18b_0
+  - urllib3=2.2.3=py310h06a4308_0
+  - uvicorn=0.30.6=py310h06a4308_0
+  - werkzeug=3.0.6=py310h06a4308_0
+  - wheel=0.44.0=py310h06a4308_0
+  - wrapt=1.14.1=py310h5eee18b_0
+  - xz=5.4.6=h5eee18b_1
+  - yarl=1.18.0=py310h5eee18b_0
+  - zlib=1.2.13=h5eee18b_1
+  - zstd=1.5.6=hc292b87_0