add new gradio ui

.gitignore

@@ -0,0 +1 @@
+__pycache__

Dockerfile

@@ -9,13 +9,12 @@ RUN poetry config virtualenvs.create false
 # Install dependencies
 WORKDIR /app
 
-RUN git clone https://github.com/fishaudio/fish-diffusion.git --depth 1 --branch v2.1.0
+RUN git clone https://github.com/fishaudio/fish-diffusion.git --depth 1
 
 WORKDIR /app/fish-diffusion
 RUN poetry install
 
-COPY --chown=user checkpoint.ckpt checkpoints/checkpoint.ckpt
-COPY --chown=user exp_hifisinger.py configs/exp_hifisinger.py
+COPY --chown=user . .
 
 ENV GRADIO_SERVER_NAME=0.0.0.0
 ENV GRADIO_SERVER_PORT=7860
@@ -24,4 +23,4 @@ ENV TRANSFORMERS_CACHE=/tmp/huggingface
 ENV MPLCONFIGDIR=/tmp/matplotlib
 ENV TORCH_HOME=/tmp/torch
 
-CMD python3 tools/hifisinger/inference.py --config configs/exp_hifisinger.py --checkpoint checkpoints/checkpoint.ckpt --speaker opencpop --max_slice_duration 120 --min_silence_duration 2 --gradio
+CMD python3 tools/hifisinger/gradio_ui.py

config.yaml

@@ -0,0 +1,18 @@
+
+readme: |
+  # Fish Diffusion - HiFiSinger Demo 🎤
+  GitHub Repo: [fishaudio/fish-diffusion](https://github.com/fishaudio/fish-diffusion)
+
+max_mixing_speakers: 3
+
+models:
+  - name: Default
+    config: configs/exp_hifisinger.py
+    checkpoint: checkpoints/checkpoint.ckpt
+    readme: "This model is trained on the M4Singer dataset and released under the CC-BY-NC-SA 4.0 license."
+    default_speaker: "aria"
+
+  - name: Duplicate
+    config: configs/exp_hifisinger.py
+    checkpoint: checkpoints/checkpoint.ckpt
+    readme: "DUP This model is trained on the M4Singer dataset and released under the CC-BY-NC-SA 4.0 license."

configs/exp_hifisinger.py

@@ -1,5 +1,5 @@
 _base_ = [
-    "./_base_/archs/hifi_svc.py",
+    "/home/lengyue/workspace/ml-toys/diff-svc-clean/configs/_base_/archs/hifi_svc.py",
 ]
 
 speaker_mapping = {'DELETED0': 0, 'opencpop': 1, 'DELETED2': 2, 'DELETED3': 3, 'M4Singer-Alto-7': 4, 'M4Singer-Alto-1': 5, 'M4Singer-Alto-5': 6, 'M4Singer-Tenor-5': 7, 'M4Singer-Alto-2': 8, 'M4Singer-Tenor-7': 9, 'M4Singer-Tenor-4': 10, 'M4Singer-Alto-6': 11, 'M4Singer-Soprano-3': 12, 'M4Singer-Bass-1': 13, 'M4Singer-Bass-3': 14, 'M4Singer-Tenor-2': 15, 'M4Singer-Alto-3': 16, 'M4Singer-Tenor-6': 17, 'M4Singer-Bass-2': 18, 'M4Singer-Alto-4': 19, 'M4Singer-Soprano-2': 20, 'M4Singer-Soprano-1': 21, 'M4Singer-Alto-2#forever': 22, 'M4Singer-Tenor-3': 23, 'M4Singer-Tenor-1': 24, 'M4Singer-Tenor-1#always': 25}