Update app.py

app.py

@@ -52,12 +52,9 @@ nest_asyncio.apply()
 SAMPLE_RATE = 16000
 SF2_PATH = 'SGM-v2.01-Sal-Guit-Bass-V1.3.sf2'
 
-def upload_audio(sample_rate):
-  data = list(files.upload().values())
-  if len(data) > 1:
-    print('Multiple files uploaded; using only one.')
+def upload_audio(audio, sample_rate):
   return note_seq.audio_io.wav_data_to_samples_librosa(
-    data[0], sample_rate=sample_rate)
+    audio, sample_rate=sample_rate)
 
 
 
@@ -259,24 +256,33 @@ class InferenceModel(object):
 
 
 
+
+
+inference_model = InferenceModel('./checkpoints/mt3/', 'mt3')
+
+
 def inference(audio):
-  os.system("midi_ddsp_synthesize --midi_path "+audio.name)
-  return Path(audio.name).stem+"/0_violin.wav"
+  audio = upload_audio(audio,sample_rate=16000)
+  
+  est_ns = inference_model(audio)
+  
+  note_seq.sequence_proto_to_midi_file(est_ns, './transcribed.mid')
+ 
+  return './transcribed.mid'
   
 title = "Midi-DDSP"
 description = "Gradio demo for MIDI-DDSP: Detailed Control of Musical Performance via Hierarchical Modeling. To use it, simply upload your midi file, or click one of the examples to load them. Read more at the links below."
 
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.09312' target='_blank'>MIDI-DDSP: Detailed Control of Musical Performance via Hierarchical Modeling</a> | <a href='https://github.com/magenta/midi-ddsp' target='_blank'>Github Repo</a></p>"
 
-examples=[['input.mid']]
+examples=[['download.wav']]
 
 gr.Interface(
     inference, 
-    gr.inputs.File(type="file", label="Input"), 
-    [gr.outputs.Audio(type="file", label="Output")],
+    gr.inputs.Audio(type="filepath", label="Input"), 
+    [gr.outputs.File(type="file", label="Output")],
     title=title,
     description=description,
     article=article,
-    examples=examples,
-    enable_queue=True
-    ).launch(debug=True)
+    examples=examples
+    ).launch(enable_queue=True)