Auto sync at 2026-05-03 16:27:53

app.py

@@ -5,27 +5,7 @@ import gradio as gr
 from mutagen.mp3 import MP3
 from mutagen.flac import FLAC
 from piano_transcription_inference import PianoTranscription, load_audio, sample_rate
-from convert import midi2xml, xml2abc, xml2mxl, xml2jpg
-
-EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
-TMP_DIR = "./__pycache__"
-
-if EN_US:
-    import huggingface_hub
-
-    MODEL_PATH = huggingface_hub.snapshot_download(
-        "Genius-Society/piano_trans",
-        cache_dir=TMP_DIR,
-    )
-
-else:
-    import modelscope
-
-    MODEL_PATH = modelscope.snapshot_download(
-        "Genius-Society/piano_trans",
-        cache_dir=TMP_DIR,
-    )
-
+from convert import midi2xml, xml2abc, xml2mxl, xml2jpg, EN_US, TMP_DIR, MODEL_DIR
 
 ZH2EN = {
     "五线谱": "Staff",
@@ -88,7 +68,7 @@ def audio2midi(audio_path: str, cache_dir: str):
     audio, _ = load_audio(audio_path, sr=sample_rate, mono=True)
     transcriptor = PianoTranscription(
         device="cuda" if torch.cuda.is_available() else "cpu",
-        checkpoint_path=f"{MODEL_PATH}/CRNN_note_F1=0.9677_pedal_F1=0.9186.pth",
+        checkpoint_path=f"{MODEL_DIR}/CRNN_note_F1=0.9677_pedal_F1=0.9186.pth",
     )
     midi_path = f"{cache_dir}/output.mid"
     transcriptor.transcribe(audio, midi_path)
@@ -112,7 +92,7 @@ def upl_infer(audio_path: str, cache_dir=f"{TMP_DIR}/cache"):
     return status, midi, pdf, xml, mxl, abc, jpg
 
 
-def find_audio_files(folder_path=f"{MODEL_PATH}/examples"):
+def find_audio_files(folder_path=f"{MODEL_DIR}/examples"):
     wav_files = []
     for root, _, files in os.walk(folder_path):
         for file in files:

convert.py

@@ -1,42 +1,35 @@
 import os
 import sys
 import fitz
-import requests
+import shutil
 import subprocess
 from PIL import Image
 from music21 import converter
 
+EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
+TMP_DIR = "./__pycache__"
 
-def download(url: str, directory: str, filename: str):
-    if directory != "" and not os.path.exists(directory):
-        os.makedirs(directory)
+if EN_US:
+    import huggingface_hub
 
-    file_path = os.path.join(directory, filename)
-    response = requests.get(url, stream=True)
-    if response.status_code == 200:
-        with open(file_path, "wb") as file:
-            for chunk in response.iter_content(chunk_size=1024):
-                if chunk:
-                    file.write(chunk)
-
-        print(f"文件已下载并保存到 {file_path}")
+    MODEL_DIR = huggingface_hub.snapshot_download(
+        "Genius-Society/piano_trans",
+        cache_dir=TMP_DIR,
+    )
 
-    else:
-        print(f"下载文件失败。状态代码: {response.status_code}")
+else:
+    import modelscope
 
-    return os.path.join(directory, filename)
+    MODEL_DIR = modelscope.snapshot_download(
+        "Genius-Society/piano_trans",
+        cache_dir=TMP_DIR,
+    )
 
 
 if sys.platform.startswith("linux"):
     apkname = "MuseScore.AppImage"
+    shutil.move(os.path.realpath(f"{MODEL_DIR}/{apkname}"), f"./{apkname}")
     extra_dir = "squashfs-root"
-    if not os.path.exists(apkname):
-        download(
-            url="https://www.modelscope.cn/studio/Genius-Society/piano_trans/resolve/master/MuseScore.AppImage",
-            directory="./",
-            filename=apkname,
-        )
-
     if not os.path.exists(extra_dir):
         subprocess.run(["chmod", "+x", f"./{apkname}"])
         subprocess.run([f"./{apkname}", "--appimage-extract"])