audio2midi 0.10.0__py2.py3-none-any.whl → 0.12.0__py2.py3-none-any.whl

audio2midi/crepe_pitch_detector.py

@@ -913,7 +913,8 @@ class Crepe():
                 inputs = batch[0].to(device)
                 outputs = self.model(inputs)
                 all_outputs.append(outputs.cpu())
-                progress_callback(i,total_batch)
+                if progress_callback:
+                    progress_callback(i,total_batch)
         return torch.cat(all_outputs, dim=0)
 
     def model_predict(self,audio:np.ndarray,viterbi, center, step_size,progress_callback,batch_size):

audio2midi/yourmt3_music_transcription.py

@@ -0,0 +1 @@
+from yourmt3 import YMT3

{audio2midi-0.10.0.dist-info → audio2midi-0.12.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: audio2midi
-Version: 0.10.0
+Version: 0.12.0
 Summary: Audio To Midi
 Author-email: dummyjenil <dummyjenil@gmail.com>
 Provides-Extra: all
@@ -88,6 +88,8 @@ Requires-Dist: pretty-midi-fix; extra == 'violin-pitch-detector'
 Requires-Dist: scipy; extra == 'violin-pitch-detector'
 Requires-Dist: torch; extra == 'violin-pitch-detector'
 Requires-Dist: torchaudio; extra == 'violin-pitch-detector'
+Provides-Extra: yourmt3-music-transcription
+Requires-Dist: yourmt3; extra == 'yourmt3-music-transcription'
 Description-Content-Type: text/markdown
 
 [Audio2Midi Demo](https://huggingface.co/spaces/shethjenil/Audio2Midi)
@@ -97,7 +99,7 @@ Description-Content-Type: text/markdown
 ---
 
 ```bash
-pip install audio2midi[all] audio2midi[pop2piano] audio2midi[violin_pitch_detector] audio2midi[crepe_pitch_detector] audio2midi[crepe_pitch_detector_tf] audio2midi[melodia_pitch_detector] audio2midi[basic_pitch_pitch_detector] audio2midi[librosa_pitch_detector] audio2midi[magenta_music_transcription]
+pip install audio2midi[all] audio2midi[pop2piano] audio2midi[violin_pitch_detector] audio2midi[crepe_pitch_detector] audio2midi[crepe_pitch_detector_tf] audio2midi[melodia_pitch_detector] audio2midi[basic_pitch_pitch_detector] audio2midi[librosa_pitch_detector] audio2midi[magenta_music_transcription] audio2midi[yourmt3_music_transcription] audio2midi[mt3_music_transcription]
 ```
 ---
 
@@ -130,9 +132,12 @@ from platform import system as platform_system , architecture as platform_archit
 
 import nest_asyncio
 from audio2midi.mt3_music_transcription import MT3
+from yourmt3_music_transcription import YMT3
 nest_asyncio.apply()
 unpack_archive(hf_hub_download("shethjenil/Audio2Midi_Models","mt3.zip"),"mt3_model",format="zip")
 MT3("mt3_model").predict(audio_path)
+name = "YMT3+"
+YMT3(hf_hub_download("shethjenil/Audio2Midi_Models",f"{name}.pt"),name,"32" if str(device) == "cpu" else "16").predict("audio.mp3")
 
 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())
@@ -168,6 +173,7 @@ 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 yourmt3_music_transcription import YMT3
 from audio2midi.magenta_music_transcription import Magenta
 from os import environ
 from huggingface_hub import hf_hub_download
@@ -229,6 +235,7 @@ gr.TabbedInterface([
     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")),
+    gr.Interface(YMT3(hf_hub_download("shethjenil/Audio2Midi_Models",f"{name}.pt"),name,"32" if str(device) == "cpu" else "16",device).predict,gr.gr.Audio(label="Input Audio",type="filepath"),gr.File(label="MIDI File"))
     midi_viz_ui
-],["Normal Pitch Detection","Guitar Based Pitch Detection","Melodia","Spotify Pitch Detection","Magenta Pitch Detection","Violin Based Pitch Detection","Crepe Pitch Detection","Crepe Pitch Detection TF","Pop2Piano","MT3","Midi Vizulizer"]).launch()
+],["Normal Pitch Detection","Guitar Based Pitch Detection","Melodia","Spotify Pitch Detection","Magenta Pitch Detection","Violin Based Pitch Detection","Crepe Pitch Detection","Crepe Pitch Detection TF","Pop2Piano","MT3","YourMT3","Midi Vizulizer"]).launch()
 ```

{audio2midi-0.10.0.dist-info → audio2midi-0.12.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 audio2midi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 audio2midi/basic_pitch_pitch_detector.py,sha256=c527_VfixXbuZYRP5uSHmoDja26ok7uc71F_HeIOXWI,30660
-audio2midi/crepe_pitch_detector.py,sha256=63PLW5pQ_WSlIr_bOSAv7wjIujA06Iy-pTQab3InPR4,34675
+audio2midi/crepe_pitch_detector.py,sha256=Gty_wDXNSnKGmq-tL17QNadihVXW40ZJ-fb7kgWLYE8,34718
 audio2midi/crepe_pitch_detector_tf.py,sha256=obOK9zWeAFWKhVuZAXG-SM2Cn-vVBjQMU08hwwjDvKA,7607
 audio2midi/librosa_pitch_detector.py,sha256=U8r98mnkkRxTlG2M0fX5Fif3qwihgfNry1pa-4lEd-E,7782
 audio2midi/magenta_music_transcription.py,sha256=5rgn_Mvmu4GClVi-dQyqyClJt58B2siPZOYNbGZuoMM,4878
@@ -9,6 +9,7 @@ audio2midi/mt3_music_transcription.py,sha256=_8MgMoXd-LNzI4GeXwIJs6pq7B7lrEkzOQw
 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.10.0.dist-info/METADATA,sha256=079whu1H7ugAMXoBlkaEpuzWRjipFxjc4CnOavaadKA,13946
-audio2midi-0.10.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-audio2midi-0.10.0.dist-info/RECORD,,
+audio2midi/yourmt3_music_transcription.py,sha256=ZS-2EOog6fVoRpbR4RjrDlHytDFNsLQOCHw344JtVsE,26
+audio2midi-0.12.0.dist-info/METADATA,sha256=9Rh6NhOea2bqMrQamK11A-53OBU9uI6S4hHLMsonE00,14596
+audio2midi-0.12.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+audio2midi-0.12.0.dist-info/RECORD,,