upd repo name

app.py

@@ -14,7 +14,7 @@ from convert import midi2xml, xml2abc, xml2mxl, xml2jpg
 
 CACHE_DIR = "./flagged"
 WEIGHTS_PATH = (
-    snapshot_download("Genius-Society/piano_transcription", cache_dir="./__pycache__")
+    snapshot_download("Genius-Society/piano_trans", cache_dir="./__pycache__")
     + "/CRNN_note_F1=0.9677_pedal_F1=0.9186.pth"
 )
 
@@ -140,6 +140,15 @@ def url_infer(audio_url: str):
 
             download_path = download_audio(audio_url, download_path)
 
+        elif audio_url.isalpha():
+            song_id = get_first_integer(audio_url)
+            audio_url = f"https://music.163.com/song/media/outer/url?id={song_id}.mp3"
+            song_name, free = music163_song_info(song_id)
+            if not free:
+                raise AttributeError("Unable to parse VIP songs")
+
+            download_path = download_audio(audio_url, download_path)
+
         midi, title = audio2midi(download_path)
         if song_name:
             title = song_name
@@ -178,7 +187,10 @@ if __name__ == "__main__":
         with gr.Tab("Direct Link Mode"):
             gr.Interface(
                 fn=url_infer,
-                inputs=gr.Textbox(label="Input audio direct link"),
+                inputs=gr.Textbox(
+                    label="Input audio direct link",
+                    placeholder="https://music.163.com/#/song?id=",
+                ),
                 outputs=[
                     gr.Audio(label="Download audio", type="filepath"),
                     gr.File(label="Download MIDI"),

convert.py

@@ -36,7 +36,7 @@ if sys.platform.startswith("linux"):
     extra_dir = "squashfs-root"
     if not os.path.exists(apkname):
         download(
-            url="https://www.modelscope.cn/studio/Genius-Society/piano_transcription/resolve/master/MuseScore.AppImage",
+            url="https://www.modelscope.cn/studio/Genius-Society/piano_trans/resolve/master/MuseScore.AppImage",
             directory="./",
             filename=apkname,
         )