sync ms

app.py

@@ -7,13 +7,17 @@ import numpy as np
 import gradio as gr
 import librosa.display
 import matplotlib.pyplot as plt
-from utils import get_modelist, find_audio_files, embed_img
 from model import EvalNet
-
-
-CLASSES = ["Gong", "Shang", "Jue", "Zhi", "Yu"]
-TEMP_DIR = "./__pycache__/tmp"
-SAMPLE_RATE = 44100
+from utils import (
+    get_modelist,
+    find_audio_files,
+    embed_img,
+    _L,
+    SAMPLE_RATE,
+    TEMP_DIR,
+    TRANSLATE,
+    CLASSES,
+)
 
 
 def zero_padding(y: np.ndarray, end: int):
@@ -73,28 +77,29 @@ def audio2chroma(audio_path: str, seg_len=20):
 
 
 def infer(wav_path: str, log_name: str, folder_path=TEMP_DIR):
-    if os.path.exists(folder_path):
-        shutil.rmtree(folder_path)
+    status = "Success"
+    filename = result = None
+    try:
+        if os.path.exists(folder_path):
+            shutil.rmtree(folder_path)
 
-    if not wav_path:
-        return None, "Please input an audio!"
+        if not wav_path:
+            raise ValueError("请输入音频!")
 
-    spec = log_name.split("_")[-3]
-    os.makedirs(folder_path, exist_ok=True)
-    try:
-        model = EvalNet(log_name, len(CLASSES)).model
+        spec = log_name.split("_")[-3]
+        os.makedirs(folder_path, exist_ok=True)
+        model = EvalNet(log_name, len(TRANSLATE)).model
         eval("audio2%s" % spec)(wav_path)
+        input = embed_img(f"{folder_path}/output.jpg")
+        output: torch.Tensor = model(input)
+        pred_id = torch.max(output.data, 1)[1]
+        filename = os.path.basename(wav_path)
+        result = f"{TRANSLATE[CLASSES[pred_id]]} ({CLASSES[pred_id].capitalize()})"
 
     except Exception as e:
-        return None, f"{e}"
-
-    input = embed_img(f"{folder_path}/output.jpg")
-    output: torch.Tensor = model(input)
-    pred_id = torch.max(output.data, 1)[1]
-    return (
-        os.path.basename(wav_path),
-        CLASSES[pred_id].capitalize(),
-    )
+        status = f"{e}"
+
+    return status, filename, result
 
 
 if __name__ == "__main__":
@@ -109,39 +114,40 @@ if __name__ == "__main__":
         gr.Interface(
             fn=infer,
             inputs=[
-                gr.Audio(label="Upload a recording", type="filepath"),
-                gr.Dropdown(choices=models, label="Select a model", value=models[0]),
+                gr.Audio(label=_L("上传录音"), type="filepath"),
+                gr.Dropdown(choices=models, label=_L("选择模型"), value=models[0]),
             ],
             outputs=[
-                gr.Textbox(label="Audio filename", show_copy_button=True),
+                gr.Textbox(label=_L("状态栏"), show_copy_button=True),
+                gr.Textbox(label=_L("音频文件名"), show_copy_button=True),
                 gr.Textbox(
-                    label="Chinese pentatonic mode recognition",
+                    label=_L("中国五声调式识别"),
                     show_copy_button=True,
                 ),
             ],
             examples=examples,
             cache_examples=False,
             flagging_mode="never",
-            title="It is recommended to keep the recording length around 20s.",
+            title=_L("建议录音时长保持在 20s 左右"),
         )
 
         gr.Markdown(
-            """
-# Cite
-```bibtex
-@article{Zhou-2025,
-  author  = {Monan Zhou and Shenyang Xu and Zhaorui Liu and Zhaowen Wang and Feng Yu and Wei Li and Baoqiang Han},
-  title   = {CCMusic: An Open and Diverse Database for Chinese Music Information Retrieval Research},
-  journal = {Transactions of the International Society for Music Information Retrieval},
-  volume  = {8},
-  number  = {1},
-  pages   = {22--38},
-  month   = {Mar},
-  year    = {2025},
-  url     = {https://doi.org/10.5334/tismir.194},
-  doi     = {10.5334/tismir.194}
-}
-```"""
+            f"# {_L('引用')}"
+            + """
+            ```bibtex
+            @article{Zhou-2025,
+                author  = {Monan Zhou and Shenyang Xu and Zhaorui Liu and Zhaowen Wang and Feng Yu and Wei Li and Baoqiang Han},
+                title   = {CCMusic: An Open and Diverse Database for Chinese Music Information Retrieval Research},
+                journal = {Transactions of the International Society for Music Information Retrieval},
+                volume  = {8},
+                number  = {1},
+                pages   = {22--38},
+                month   = {Mar},
+                year    = {2025},
+                url     = {https://doi.org/10.5334/tismir.194},
+                doi     = {10.5334/tismir.194}
+            }
+            ```"""
         )
 
-    demo.launch(ssr_mode=False)
+    demo.launch()

model.py

@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 import torchvision.models as models
-from datasets import load_dataset
+from modelscope.msdatasets import MsDataset
 from utils import MODEL_DIR
 
 
@@ -17,7 +17,7 @@ class EvalNet:
         self.m_type, self.input_size = self._model_info(m_ver)
 
         if not hasattr(models, m_ver):
-            raise Exception("Unsupported model.")
+            raise Exception("不支持的模型")
 
         self.model = eval("models.%s()" % m_ver)
         linear_output = self._set_outsize()
@@ -34,11 +34,14 @@ class EvalNet:
             if ver == bb["ver"]:
                 return bb
 
-        print("Backbone name not found, using default option - alexnet.")
+        print("未找到骨干网络名称，使用默认选项 - alexnet")
         return backbone_list[0]
 
     def _model_info(self, m_ver: str):
-        backbone_list = load_dataset("monetjoe/cv_backbones", split="train")
+        backbone_list = MsDataset.load(
+            "monetjoe/cv_backbones",
+            split="v1",
+        )
         backbone = self._get_backbone(m_ver, backbone_list)
         m_type = str(backbone["type"])
         input_size = int(backbone["input_size"])

requirements.txt

@@ -1,5 +1,7 @@
-torch
-pillow
+torch==2.6.0+cu118
+-f https://download.pytorch.org/whl/torch
+torchvision==0.21.0+cu118
+-f https://download.pytorch.org/whl/torchvision
 librosa
 matplotlib
-torchvision
+modelscope[framework]==1.21.0

utils.py

@@ -1,10 +1,54 @@
 import os
 import torch
 import torchvision.transforms as transforms
-from huggingface_hub import snapshot_download
+import huggingface_hub
+import modelscope
 from PIL import Image
 
-MODEL_DIR = snapshot_download("ccmusic-database/CNPM", cache_dir="./__pycache__")
+EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
+
+ZH2EN = {
+    "上传录音": "Upload a recording",
+    "选择模型": "Select a model",
+    "状态栏": "Status",
+    "音频文件名": "Audio filename",
+    "中国五声调式识别": "Chinese pentatonic mode recognition",
+    "建议录音时长保持在 20s 左右": "It is recommended to keep the recording length around 20s.",
+    "引用": "Cite",
+    "宫": "Gong",
+    "商": "Shang",
+    "角": "Jue",
+    "徵": "Zhi",
+    "羽": "Yu",
+}
+
+MODEL_DIR = (
+    huggingface_hub.snapshot_download(
+        "ccmusic-database/CNPM",
+        cache_dir="./__pycache__",
+    )
+    if EN_US
+    else modelscope.snapshot_download(
+        "ccmusic-database/CNPM",
+        cache_dir="./__pycache__",
+    )
+)
+
+
+def _L(zh_txt: str):
+    return ZH2EN[zh_txt] if EN_US else zh_txt
+
+
+TRANSLATE = {
+    "Gong": _L("宫"),
+    "Shang": _L("商"),
+    "Jue": _L("角"),
+    "Zhi": _L("徵"),
+    "Yu": _L("羽"),
+}
+CLASSES = list(TRANSLATE.keys())
+TEMP_DIR = "./__pycache__/tmp"
+SAMPLE_RATE = 44100
 
 
 def toCUDA(x):
@@ -27,19 +71,16 @@ def find_audio_files(folder_path=f"{MODEL_DIR}/examples"):
 
 
 def get_modelist(model_dir=MODEL_DIR, assign_model=""):
-    try:
-        entries = os.listdir(model_dir)
-    except OSError as e:
-        print(f"Cannot access {model_dir}: {e}")
-        return
-
     output = []
-    for entry in entries:
+    for entry in os.listdir(model_dir):
+        # 获取完整路径
         full_path = os.path.join(model_dir, entry)
+        # 跳过'.git'文件夹
         if entry == ".git" or entry == "examples":
-            print(f"Skip .git / examples dir: {full_path}")
+            print(f"跳过 .git 或 examples 文件夹: {full_path}")
             continue
 
+        # 检查条目是文件还是目录
         if os.path.isdir(full_path):
             model = os.path.basename(full_path)
             if assign_model and assign_model.lower() in model: