From ad33f65138743f31ad2dea2a9183bf01feec0c03 Mon Sep 17 00:00:00 2001
From: lovegaoshi <106490582+lovegaoshi@users.noreply.github.com>
Date: Fri, 29 Nov 2024 14:30:32 +0000
Subject: [PATCH] fix: tensorflow 2.18.0

---
 Dockerfile       | 12 +++++++-----
 Dockerfile-gpu   | 11 +++++++----
 requirements.txt |  1 -
 3 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index df99513..58876f1 100644
--- 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
index cf8dd49..3bbeb73 100644
--- 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
index 169b025..a36aa0c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,4 @@
 yt-dlp
-inaspeechsegmenter
 shazamio
 pyyaml
 filelock