changed dockerfile to the docker hub version

This commit is contained in:
Jaikinator
2023-09-29 13:37:16 +02:00
parent ef7a4273e6
commit 98f41beab5
+21 -3
View File
@@ -1,5 +1,17 @@
#pytorch Image #pytorch Image
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime 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 # Install dependencies
WORKDIR /app WORKDIR /app
ARG hf_token ARG hf_token
@@ -10,17 +22,23 @@ ENV AUTOT_CACHE /app/models
ENV PYANNOTE_CACHE /app/models/pyannote ENV PYANNOTE_CACHE /app/models/pyannote
#Copy all necessary files #Copy all necessary files
COPY requirements.txt /app/requirements.txt 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 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
RUN conda install pip RUN conda install pip
RUN conda install -y ffmpeg RUN conda install -y ffmpeg
RUN conda install -c conda-forge libsndfile 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 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install /app/ RUN pip install /app/
RUN pip install markupsafe==2.0.1 --force-reinstall RUN pip install markupsafe==2.0.1 --force-reinstall
RUN Scraibe --hf_token $hf_token RUN scraibe --whisper-model-name small
# Expose port # Expose port
EXPOSE 7860 EXPOSE 7860
# Run the application # Run the application
ENTRYPOINT ["scraibe"] ENTRYPOINT ["scraibe" ,"--whisper-model-name", "small"]