Bring dockerfile up to date

This commit is contained in:
Marko Henning
2024-10-10 14:07:36 +02:00
parent ce2f3ebde2
commit 08f14883e2
+18 -20
View File
@@ -1,5 +1,5 @@
#pytorch Image #pytorch Image
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
# Labels # Labels
@@ -14,33 +14,31 @@ LABEL url="https://github.com/JSchmie/ScrAIbe"
# Install dependencies # Install dependencies
WORKDIR /app WORKDIR /app
ARG model_name=medium #Enviorment dependencies
#Enviorment Dependncies ENV TRANSFORMERS_CACHE=/app/models
ENV TRANSFORMERS_CACHE /app/models ENV HF_HOME=/app/models
ENV HF_HOME /app/models ENV AUTOT_CACHE=/app/models
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 README.md /app/README.md COPY README.md /app/README.md
COPY models /app/models
COPY scraibe /app/scraibe 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 apt update -y && apt upgrade -y && \
RUN conda update --all apt install -y libsm6 libxrender1 libfontconfig1 && \
apt clean && \
rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
RUN conda install pip RUN conda update --all && \
RUN conda install -y ffmpeg # conda install -y pip ffmpeg && \
RUN conda install -c conda-forge libsndfile conda install -c conda-forge libsndfile && \
RUN pip install torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html conda clean --all -y
RUN pip install -r requirements.txt # RUN pip install torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install markupsafe==2.0.1 --force-reinstall RUN pip install --no-cache-dir -r requirements.txt
RUN python3 -m 'scraibe.cli' --whisper-model-name $model_name
# Expose port # Expose port
EXPOSE 7860 EXPOSE 7860
# Run the application # Run the application
ENTRYPOINT ["python3", "-m", "scraibe.cli" ,"--whisper-model-name", "$model_name"] ENTRYPOINT ["python3", "-m", "scraibe.cli"]