audio2midi 0.4.0__py2.py3-none-any.whl → 0.5.0__py2.py3-none-any.whl

audio2midi/mt3_music_transcription.py

@@ -0,0 +1 @@
+from mt3_audio2midi import MT3

{audio2midi-0.4.0.dist-info → audio2midi-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: audio2midi
-Version: 0.4.0
+Version: 0.5.0
 Summary: Audio To Midi
 Author-email: dummyjenil <dummyjenil@gmail.com>
 Provides-Extra: all
@@ -10,6 +10,8 @@ Requires-Dist: huggingface-hub; extra == 'all'
 Requires-Dist: keras; extra == 'all'
 Requires-Dist: librosa; extra == 'all'
 Requires-Dist: mir-eval; extra == 'all'
+Requires-Dist: mt3-audio2midi; extra == 'all'
+Requires-Dist: nest-asyncio; extra == 'all'
 Requires-Dist: nnaudio; extra == 'all'
 Requires-Dist: numpy==1.26.4; extra == 'all'
 Requires-Dist: pretty-midi; extra == 'all'
@@ -57,6 +59,9 @@ Requires-Dist: numpy; extra == 'melodia-pitch-detector'
 Requires-Dist: pretty-midi-fix; extra == 'melodia-pitch-detector'
 Requires-Dist: scipy; extra == 'melodia-pitch-detector'
 Requires-Dist: vamp; extra == 'melodia-pitch-detector'
+Provides-Extra: mt3-music-transcription
+Requires-Dist: mt3-audio2midi; extra == 'mt3-music-transcription'
+Requires-Dist: nest-asyncio; extra == 'mt3-music-transcription'
 Provides-Extra: pop2piano
 Requires-Dist: essentia; extra == 'pop2piano'
 Requires-Dist: huggingface-hub; extra == 'pop2piano'
@@ -117,6 +122,12 @@ from pathlib import Path
 from audio2midi.melodia_pitch_detector import Melodia
 from platform import system as platform_system , architecture as platform_architecture
 
+import nest_asyncio
+from audio2midi.mt3_music_transcription import MT3
+nest_asyncio.apply()
+unpack_archive(hf_hub_download("shethjenil/Audio2Midi_Models","mt3.zip"),"mt3_model",format="zip")
+MT3("mt3_model").predict(audio_path)
+
 unpack_archive(hf_hub_download("shethjenil/Audio2Midi_Models",f"melodia_vamp_plugin_{'win' if (system := platform_system()) == 'Windows' else 'mac' if system == 'Darwin' else 'linux64' if (arch := platform_architecture()[0]) == '64bit' else 'linux32' if arch == '32bit' else None}.zip"),"vamp_melodia",format="zip")
 environ['VAMP_PATH'] = str(Path("vamp_melodia").absolute())
 Melodia().predict(audio_path)
@@ -148,13 +159,18 @@ from audio2midi.librosa_pitch_detector import Normal_Pitch_Det , Guitar_Pitch_De
 from audio2midi.melodia_pitch_detector import Melodia
 from audio2midi.pop2piano import Pop2Piano
 from audio2midi.violin_pitch_detector import Violin_Pitch_Det
-
+from audio2midi.mt3_music_transcription import MT3
 from os import environ
 from huggingface_hub import hf_hub_download
 from shutil import unpack_archive
 from pathlib import Path
 from platform import system as platform_system , architecture as platform_architecture
+import nest_asyncio
+nest_asyncio.apply()
+
 unpack_archive(hf_hub_download("shethjenil/Audio2Midi_Models",f"melodia_vamp_plugin_{'win' if (system := platform_system()) == 'Windows' else 'mac' if system == 'Darwin' else 'linux64' if (arch := platform_architecture()[0]) == '64bit' else 'linux32' if arch == '32bit' else None}.zip"),"vamp_melodia",format="zip")
+unpack_archive(hf_hub_download("shethjenil/Audio2Midi_Models","mt3.zip"),"mt3_model",format="zip")
+
 environ['VAMP_PATH'] = str(Path("vamp_melodia").absolute())
 
 from os import getenv
@@ -202,6 +218,7 @@ gr.TabbedInterface([
     gr.Interface(Crepe(getenv("crepe_model_capacity","full")).predict,[gr.Audio(type="filepath",label="Input Audio"),gr.Checkbox(False,label="viterbi",info="Apply viterbi smoothing to the estimated pitch curve"),gr.Checkbox(True,label="center"),gr.Number(10,label="step size",info="The step size in milliseconds for running pitch estimation."),gr.Number(0.8,label="minimum confidence"),gr.Number(32,label="batch size")],gr.File(label="Midi File")),
     gr.Interface(CrepeTF(getenv("crepe_model_capacity","full")).predict,[gr.Audio(type="filepath",label="Input Audio"),gr.Checkbox(False,label="viterbi",info="Apply viterbi smoothing to the estimated pitch curve"),gr.Checkbox(True,label="center"),gr.Number(10,label="step size",info="The step size in milliseconds for running pitch estimation."),gr.Number(0.8,label="minimum confidence"),gr.Number(32,label="batch size")],gr.File(label="Midi File")),
     gr.Interface(Pop2Piano(device).predict,[gr.Audio(label="Input Audio",type="filepath"),gr.Number(1, minimum=1, maximum=21, label="Composer"),gr.Number(2,label="Details in Piano"),gr.Number(1,label="Efficiency of Piano"),gr.Radio([1,2,4],label="steps per beat",value=2)],gr.File(label="MIDI File")),
+    gr.Interface(MT3(str(Path("mt3_model").absolute())).predict,[gr.Audio(label="Input Audio",type="filepath"),gr.Number(0,label="seed")],gr.File(label="MIDI File")),
     midi_viz_ui
-],["Normal Pitch Detection","Guitar Based Pitch Detection","Melodia","Spotify Pitch Detection","Violin Based Pitch Detection","Crepe Pitch Detection","Crepe Pitch Detection TF","Pop2Piano","Midi Vizulizer"]).launch()
+],["Normal Pitch Detection","Guitar Based Pitch Detection","Melodia","Spotify Pitch Detection","Violin Based Pitch Detection","Crepe Pitch Detection","Crepe Pitch Detection TF","Pop2Piano","MT3","Midi Vizulizer"]).launch()
 ```

{audio2midi-0.4.0.dist-info → audio2midi-0.5.0.dist-info}/RECORD

@@ -4,9 +4,10 @@ audio2midi/crepe_pitch_detector.py,sha256=63PLW5pQ_WSlIr_bOSAv7wjIujA06Iy-pTQab3
 audio2midi/crepe_pitch_detector_tf.py,sha256=obOK9zWeAFWKhVuZAXG-SM2Cn-vVBjQMU08hwwjDvKA,7607
 audio2midi/librosa_pitch_detector.py,sha256=btwYkNqVgkf9rdkBtH2Q5DqdNVKtO9iZ88SvcbQK1jk,7777
 audio2midi/melodia_pitch_detector.py,sha256=YFt9NKyZ_Dyt_3ltInUz6QZgovmmPK6gR9EPlL_aV5Y,2402
+audio2midi/mt3_music_transcription.py,sha256=_8MgMoXd-LNzI4GeXwIJs6pq7B7lrEkzOQwQSPTo1wo,32
 audio2midi/pop2piano.py,sha256=kBAF1kY-5Ctu92etNLo4Clr1hkW1B5OCvd-XT7SrI8g,125685
 audio2midi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 audio2midi/violin_pitch_detector.py,sha256=C0--R0NaYcIyAFELGgebbs3yjAnQeVdtpk4iZ2NbUIA,61650
-audio2midi-0.4.0.dist-info/METADATA,sha256=n3x7UbE3N0APhY1HhqyU6QKbOgRTcl5sY6h2_ZTVU6M,12003
-audio2midi-0.4.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-audio2midi-0.4.0.dist-info/RECORD,,
+audio2midi-0.5.0.dist-info/METADATA,sha256=V9B8lUncLjTxdWvM25lY2RCTTmqTwagsEOrjGP2o7z8,12857
+audio2midi-0.5.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+audio2midi-0.5.0.dist-info/RECORD,,