audio2midi 0.4.0__tar.gz → 0.5.0__tar.gz

{audio2midi-0.4.0 → audio2midi-0.5.0}/PKG-INFO

@@ -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 → audio2midi-0.5.0}/README.md

@@ -36,6 +36,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)
@@ -67,13 +73,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
@@ -121,6 +132,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 → audio2midi-0.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "audio2midi"
-version = "0.4.0"
+version = "0.5.0"
 description = "Audio To Midi"
 readme = "README.md"
 authors = [
@@ -17,7 +17,8 @@ crepe_pitch_detector_tf = ["librosa", "numpy","pretty_midi_fix","hmmlearn","tens
 crepe_pitch_detector = ["librosa", "numpy","pretty_midi_fix","hmmlearn","tensorflow","torch","huggingface_hub","tqdm"]
 violin_pitch_detector = ["librosa", "numpy","pretty_midi_fix","scipy","torchaudio","torch","mir_eval","huggingface_hub"]
 pop2piano = ["librosa", "numpy==1.26.4","pretty_midi_fix","transformers","essentia","torch","scipy","resampy","pretty_midi","huggingface_hub"]
-all = ["librosa", "numpy==1.26.4","pretty_midi_fix","transformers","essentia","torch","scipy","torchaudio","torch","mir_eval","hmmlearn","tensorflow","keras","vamp","nnAudio","resampy","pretty_midi","huggingface_hub"]
+mt3_music_transcription = ["nest_asyncio", "mt3-audio2midi"]
+all = ["librosa", "numpy==1.26.4","pretty_midi_fix","transformers","essentia","torch","scipy","torchaudio","torch","mir_eval","hmmlearn","tensorflow","keras","vamp","nnAudio","resampy","pretty_midi","huggingface_hub","nest_asyncio", "mt3-audio2midi"]
 
 [build-system]
 requires = ["hatchling"]

audio2midi-0.5.0/src/audio2midi/mt3_music_transcription.py

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