Create Dockerfile

Dockerfile

@@ -0,0 +1,64 @@
+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
+
+COPY 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"]
+
+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"]