✍️ move dependencies to requirements.txt

Author: nglehuy

Dockerfile

@@ -1,9 +1,9 @@
-FROM tensorflow/tensorflow:2.3.0-gpu
+FROM tensorflow/tensorflow:2.3.2-gpu
 
 RUN apt-get update \
-&& apt-get upgrade -y \
-&& apt-get install -y \
-&& apt-get -y install apt-utils gcc libpq-dev libsndfile-dev git build-essential cmake screen
+    && apt-get upgrade -y \
+    && apt-get install -y \
+    && apt-get -y install apt-utils gcc libpq-dev libsndfile-dev git build-essential cmake screen
 
 # Clear cache
 RUN apt clean && apt-get clean
@@ -17,8 +17,8 @@ COPY scripts /scripts
 ARG install_rnnt_loss=true
 ARG using_gpu=true
 RUN if [ "$install_rnnt_loss" = "true" ] ; \
-then if [ "$using_gpu" = "true" ] ; then export CUDA_HOME=/usr/local/cuda ; else echo 'Using CPU' ; fi \
-&& ./scripts/install_rnnt_loss.sh \
-else echo 'Using pure TensorFlow'; fi
+    then if [ "$using_gpu" = "true" ] ; then export CUDA_HOME=/usr/local/cuda ; else echo 'Using CPU' ; fi \
+    && ./scripts/install_rnnt_loss.sh \
+    else echo 'Using pure TensorFlow'; fi
 
 RUN echo "export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}" >> /root/.bashrc

MANIFEST.in

@@ -0,0 +1 @@
+include requirements.txt

requirements.txt

@@ -4,7 +4,7 @@ scipy
 sklearn
 pandas
 tensorflow-datasets>=4.2.0
-tensorflow-addons>=0.10.0
+tensorflow-addons>=0.11.1
 tensorflow-io>=0.16.0
 setuptools>=47.1.1
 librosa>=0.8.0

setup.py

@@ -17,22 +17,8 @@
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-requirements = [
-    "tensorflow-datasets>=4.2.0",
-    "tensorflow-addons>=0.10.0",
-    "tensorflow-io>=0.16.0",
-    "setuptools>=47.1.1",
-    "librosa>=0.8.0",
-    "soundfile>=0.10.3",
-    "PyYAML>=5.3.1",
-    "matplotlib>=3.2.1",
-    "sox>=1.4.1",
-    "tqdm>=4.54.1",
-    "colorama>=0.4.4",
-    "nlpaug>=1.1.1",
-    "nltk>=3.5",
-    "sentencepiece>=0.1.94"
-]
+with open("requirements.txt", "r") as fr:
+    requirements = fr.read().splitlines()
 
 setuptools.setup(
     name="TensorFlowASR",