From 73dd9e7f11f864c6ce64620ca1ab6521bf97da5c Mon Sep 17 00:00:00 2001 From: Jacob Schmieder Date: Tue, 26 Sep 2023 10:00:46 +0200 Subject: [PATCH 1/3] Delete scraibe/.pyannotetoken --- scraibe/.pyannotetoken | 1 - 1 file changed, 1 deletion(-) delete mode 100644 scraibe/.pyannotetoken diff --git a/scraibe/.pyannotetoken b/scraibe/.pyannotetoken deleted file mode 100644 index 42ba269..0000000 --- a/scraibe/.pyannotetoken +++ /dev/null @@ -1 +0,0 @@ -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 \ No newline at end of file From 98f41beab58430bed9fe25afa264cb3cdd656dda Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Fri, 29 Sep 2023 13:37:16 +0200 Subject: [PATCH 2/3] changed dockerfile to the docker hub version --- Dockerfile | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 154982b..53583d1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,17 @@ #pytorch Image FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime + +# Labels + +LABEL maintainer="Jacob Schmieder" +LABEL email="Jacob.Schmieder@dbfz.de" +LABEL version="0.1.2" +LABEL description="Scraibe is a tool for automatic speech recognition and speaker diarization. \ + It is based on the Hugging Face Transformers library and the Pyannote library. \ + It is designed to be used with the Whisper model, a lightweight model for automatic \ + speech recognition and speaker diarization." +LABEL url="https://github.com/JSchmie/ScrAIbe" + # Install dependencies WORKDIR /app ARG hf_token @@ -10,17 +22,23 @@ ENV AUTOT_CACHE /app/models ENV PYANNOTE_CACHE /app/models/pyannote #Copy all necessary files COPY requirements.txt /app/requirements.txt -COPY scraibe /app/Scraibe +COPY README.md /app/README.md +COPY models /app/models +COPY scraibe /app/scraibe COPY setup.py /app/setup.py #Installing all necessary Dependencies and Running the Application with a personalised Hugging-Face-Token + +RUN apt update && apt-get install -y libsm6 libxrender1 libfontconfig1 +RUN conda update --all + RUN conda install pip RUN conda install -y ffmpeg RUN conda install -c conda-forge libsndfile RUN pip install torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html RUN pip install /app/ RUN pip install markupsafe==2.0.1 --force-reinstall -RUN Scraibe --hf_token $hf_token +RUN scraibe --whisper-model-name small # Expose port EXPOSE 7860 # Run the application -ENTRYPOINT ["scraibe"] +ENTRYPOINT ["scraibe" ,"--whisper-model-name", "small"] From 994dfadf178f631dcfaa9beb3b0537535b6d094d Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Wed, 4 Oct 2023 14:59:15 +0200 Subject: [PATCH 3/3] updated docker file --- Dockerfile | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/Dockerfile b/Dockerfile index 53583d1..95db093 100644 --- a/Dockerfile +++ b/Dockerfile @@ -5,7 +5,7 @@ FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime LABEL maintainer="Jacob Schmieder" LABEL email="Jacob.Schmieder@dbfz.de" -LABEL version="0.1.2" +LABEL version="0.1.1.dev" LABEL description="Scraibe is a tool for automatic speech recognition and speaker diarization. \ It is based on the Hugging Face Transformers library and the Pyannote library. \ It is designed to be used with the Whisper model, a lightweight model for automatic \ @@ -14,7 +14,7 @@ LABEL url="https://github.com/JSchmie/ScrAIbe" # Install dependencies WORKDIR /app -ARG hf_token +ARG model_name=medium #Enviorment Dependncies ENV TRANSFORMERS_CACHE /app/models ENV HF_HOME /app/models @@ -26,8 +26,8 @@ COPY README.md /app/README.md COPY models /app/models COPY scraibe /app/scraibe COPY setup.py /app/setup.py -#Installing all necessary Dependencies and Running the Application with a personalised Hugging-Face-Token +#Installing all necessary Dependencies and Running the Application with a personalised Hugging-Face-Token RUN apt update && apt-get install -y libsm6 libxrender1 libfontconfig1 RUN conda update --all @@ -35,10 +35,12 @@ RUN conda install pip RUN conda install -y ffmpeg RUN conda install -c conda-forge libsndfile RUN pip install torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html -RUN pip install /app/ +RUN pip install -r requirements.txt RUN pip install markupsafe==2.0.1 --force-reinstall -RUN scraibe --whisper-model-name small + +RUN python3 -m 'scraibe.cli' --whisper-model-name $model_name # Expose port EXPOSE 7860 # Run the application -ENTRYPOINT ["scraibe" ,"--whisper-model-name", "small"] + +ENTRYPOINT ["python3", "-m", "scraibe.cli" ,"--whisper-model-name", "$model_name"] \ No newline at end of file