michael/AI-Profile-Router
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add speech noise separation

Browse Source
This commit is contained in:
Mikei386
2026-09-09 08:24:12 +02:00
parent 0fd1966ae3
commit 805228abfd
7 changed files with 135 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
experiments/bs-roformer-vocal-separation/Dockerfile
+10 -2
View File
@@ -11,12 +11,20 @@ RUN python -m pip install --no-cache-dir \
"python-multipart==0.0.20" \
"uvicorn[standard]==0.35.0"
# audio-separator 0.47 requires NumPy 2 while ClearVoice 0.1.2 still pins
# NumPy 1.x. Keep ClearVoice in a small overlay venv but share the image's
# CUDA-enabled PyTorch installation instead of duplicating it.
RUN python -m venv --system-site-packages /opt/clearvoice-venv \
&& /opt/clearvoice-venv/bin/python -m pip install --no-cache-dir \
"clearvoice==0.1.2" \
"numpy>=1.24.3,<2.0"
WORKDIR /app
COPY app.py index.html ./
COPY app.py index.html speech_enhance.py ./
ENV MODEL_FILENAME=model_bs_roformer_ep_317_sdr_12.9755.ckpt \
MODEL_DIR=/models \
JOB_DIR=/data/jobs
EXPOSE 8080
CMD ["sh", "-c", "for model in \"$MODEL_FILENAME\" htdemucs_ft.yaml htdemucs_6s.yaml; do audio-separator --model_filename \"$model\" --model_file_dir \"$MODEL_DIR\" --download_model_only || exit 1; done; exec uvicorn app:app --host 0.0.0.0 --port 8080 --workers 1"]
CMD ["sh", "-c", "mkdir -p \"$MODEL_DIR/clearvoice\" && ln -sfn \"$MODEL_DIR/clearvoice\" /app/checkpoints && for model in \"$MODEL_FILENAME\" htdemucs_ft.yaml htdemucs_6s.yaml; do audio-separator --model_filename \"$model\" --model_file_dir \"$MODEL_DIR\" --download_model_only || exit 1; done; /opt/clearvoice-venv/bin/python /app/speech_enhance.py --download-only || exit 1; exec uvicorn app:app --host 0.0.0.0 --port 8080 --workers 1"]