Update midi_to_colab_audio.py

midi_to_colab_audio.py

@@ -3021,13 +3021,13 @@ def midi_to_colab_audio(midi_file,
         if max_val != 0:
             ss = (ss / max_val) * np.iinfo(np.int16).max
     ss = ss.astype(np.int16)
+    
+    if output_for_gradio:
+        return ss
 
     ss = ss.swapaxes(1, 0)
 
     raw_audio = normalize_volume(ss, volume_scale)
-
-    if output_for_gradio:
-      raw_audio = np.transpose(raw_audio)
     
     return raw_audio