Spaces:
Running
Running
File size: 2,351 Bytes
70faeef 6e5c8e2 70faeef 8f9049b a44d0ec 8f9049b 70faeef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
FROM python:3.8-slim
# src https://github.com/sakemin/SOME/blob/main/Dockerfile
# Build argument to specify whether to install PyTorch with CUDA support
ARG WITH_CUDA=0
WORKDIR /app
RUN apt-get update && apt-get install -y \
build-essential \
libsndfile1 \
ffmpeg \
wget \
unzip \
git \
&& rm -rf /var/lib/apt/lists/*
RUN git clone https://github.com/sakemin/SOME && mv SOME/* .
RUN pip install --upgrade pip==24.0 && \
if [ "$WITH_CUDA" = "1" ]; then \
pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cu118; \
else \
pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu; \
fi && \
pip install -r requirements.txt
# Create directory for pretrained models and download model checkpoints
RUN mkdir -p pretrained && \
cd pretrained && \
# Download SOMEv0.0.1 model
wget -q https://github.com/openvpi/SOME/releases/download/v0.0.1/0918_continuous256_clean_3spk_fixmel.zip && \
unzip 0918_continuous256_clean_3spk_fixmel.zip && \
rm 0918_continuous256_clean_3spk_fixmel.zip && \
# Download SOMEv1 model
wget -q https://github.com/openvpi/SOME/releases/download/v1.0.0-baseline/0119_continuous128_5spk.zip && \
unzip 0119_continuous128_5spk.zip && \
rm 0119_continuous128_5spk.zip && \
# Download RMVPE model
wget -q https://github.com/yxlllc/RMVPE/releases/download/230917/rmvpe.zip && \
unzip rmvpe.zip && \
mkdir -p rmvpe && \
mv model.pt rmvpe/model.pt && \
rm rmvpe.zip
# Add volume mount points for input and output files
VOLUME ["/app/input", "/app/output"]
# Create a directory for caches and make it writable
RUN mkdir -p /app/cache && chmod -R 777 /app/cache
# Set environment variables to point to this new directory
ENV NUMBA_CACHE_DIR=/app/cache/numba_cache
ENV MPLCONFIGDIR=/app/cache/matplotlib_cache
RUN mkdir -m 700 flagged
CMD ["python", "webui.py", "--work_dir", "pretrained", "--addr", "0.0.0.0", "--port", "7860"]
# # Set entrypoint to run inference
# ENTRYPOINT ["python", "infer.py"]
# # Default command (can be overridden)
# CMD ["--help"]
# COPY --chown=user ./requirements.txt requirements.txt
# RUN pip install --no-cache-dir --upgrade -r requirements.txt
# COPY --chown=user . /app
# CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|