Update app.py

app.py

@@ -2,17 +2,26 @@ import gradio as gr
 import numpy as np
 import pretty_midi
 from accompaniment_generator.generator.base import Generator
+import os
+import uuid
+os.system("apt install fluidsynth")
+os.system("cp /usr/share/sounds/sf2/FluidR3_GM.sf2 ./font.sf2")
+
+from midi2audio import FluidSynth
+
 
 
 def inference(audio, num_epoch):
     generator = Generator()
     input_midi_data = pretty_midi.PrettyMIDI(audio.name)
     output_midi_data = generator(audio.name, num_epoch=int(num_epoch))
-    data = input_midi_data.synthesize()
-    input_scaled = np.int16(data / np.max(np.abs(data)) * 32767)
-    data = output_midi_data.synthesize()
-    output_scaled = np.int16(data / np.max(np.abs(data)) * 32767)
-    return [(44100, input_scaled), (44100, output_scaled)]
+    name = uuid.uuid4()
+    output_midi_data.write(f'{name}.mid')
+    fs = FluidSynth("font.sf2")
+    fs.midi_to_audio(f'{name}.mid', f'{name}.wav')
+    fs.midi_to_audio(audio.name, f'{name}-init.wav')
+
+    return [f'{name}-init.wav', f'{name}.wav']
 
 
 title = "Accompaniment Generator"