omnizart / Dockerfile
vedmistry's picture
fix: pin keras==2.12.0 to match TF version and support SavedModel checkpoints
2fdbbfb
Raw
History Blame Contribute Delete
1.13 kB
FROM python:3.10-slim
WORKDIR /app
RUN apt-get update && \
apt-get install -y --no-install-recommends \
git gcc libc6-dev build-essential \
libsndfile1 ffmpeg fluidsynth \
wget curl portaudio19-dev && \
rm -rf /var/lib/apt/lists/*
COPY constraints.txt /app/constraints.txt
ENV PIP_NO_BUILD_ISOLATION=1
ENV PIP_CONSTRAINT=/app/constraints.txt
ENV TF_ENABLE_ONEDNN_OPTS=0
ENV TF_CPP_MIN_LOG_LEVEL=3
RUN pip install --no-cache-dir -U pip wheel setuptools==80.9.0
RUN pip install --no-cache-dir numpy==1.23.5 Cython
RUN pip install --no-cache-dir protobuf==4.25.9
RUN pip install --no-cache-dir keras==2.12.0
RUN pip install --no-cache-dir tensorflow-cpu==2.12.0
RUN pip install --no-cache-dir omnizart
RUN PIP_CONSTRAINT="" pip install --no-cache-dir git+https://github.com/TEAMuP-dev/pyharp.git@v0.3.0
RUN pip install --no-cache-dir gradio==5.28.0
RUN pip install --no-cache-dir --force-reinstall protobuf==4.25.9 numpy==1.23.5 keras==2.12.0
RUN omnizart download-checkpoints
COPY . /app
ENV PORT=7860
EXPOSE 7860
ENV PYTHONUNBUFFERED=1
ENV PYTHONIOENCODING=UTF-8
CMD ["python", "-u", "app.py"]