fix: tensorflow 2.18.0

lovegaoshi · Nov 29, 2024 · ad33f65 · ad33f65
1 parent df5066e
commit ad33f65
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 10 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,9 +1,11 @@
-FROM tensorflow/tensorflow:2.13.0
+FROM tensorflow/tensorflow:2.18.0
 RUN apt update
-RUN apt -y install ffmpeg aria2 wget nano
+RUN apt -y install ffmpeg aria2 wget nano git
 WORKDIR /inaseg
 COPY ./requirements.txt /inaseg/requirements.txt
-RUN pip install -r /inaseg/requirements.txt
-RUN pip install --force-reinstall git+https://github.com/grqz/yt-dlp.git@ie/bilibili/pi_fallbk
+RUN git clone https://github.com/ina-foss/inaSpeechSegmenter.git
+RUN cd inaSpeechSegmenter; pip install .
+RUN pip3 install -r /inaseg/requirements.txt
+RUN pip3 install --force-reinstall git+https://github.com/grqz/yt-dlp.git@ie/bilibili/pi_fallbk
 COPY . /inaseg
-RUN python /inaseg/biliupinit.py
+RUN python3 /inaseg/biliupinit.py
diff --git a/Dockerfile-gpu b/Dockerfile-gpu
@@ -1,8 +1,11 @@
-FROM tensorflow/tensorflow:2.13.0-gpu
+FROM tensorflow/tensorflow:2.18.0-gpu
 RUN apt update
-RUN apt -y install ffmpeg aria2 wget nano
+RUN apt -y install ffmpeg aria2 wget nano git
 WORKDIR /inaseg
 COPY ./requirements.txt /inaseg/requirements.txt
-RUN pip install -r /inaseg/requirements.txt
+RUN git clone https://github.com/ina-foss/inaSpeechSegmenter.git
+RUN cd inaSpeechSegmenter; pip install .
+RUN pip3 install -r /inaseg/requirements.txt
+RUN pip3 install --force-reinstall git+https://github.com/grqz/yt-dlp.git@ie/bilibili/pi_fallbk
 COPY . /inaseg
-RUN python /inaseg/biliupinit.py
+RUN python3 /inaseg/biliupinit.py
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,4 @@
 yt-dlp
-inaspeechsegmenter
 shazamio
 pyyaml
 filelock