Added import statements

audio_methods.py

@@ -1,3 +1,17 @@
+# Flow and data
+import numpy as np
+import pandas as pd
+
+# Audio
+import pretty_midi
+
+# Displaying
+from IPython import display
+
+# Extras
+import collections
+
+
 _CAP = 3501 # Cap for the number of notes
 _SAMPLING_RATE = 16000 # Parameter to pass continuous signal to a discrete one
 _INSTRUMENT_NAME = "Acoustic Grand Piano" # MIDI instrument used
@@ -74,7 +88,6 @@ def map_to_wav(song_map: pd.DataFrame, out_file: str, velocity: int=100):
 
 def generate_and_display(out_file, model, z_sample=None, velocity=100, seconds=120):
   song_map = model.generate(z_sample)
-  display.display(imshow(tf.squeeze(song_map)[:,:50]))
   wav = map_to_wav(song_map, out_file, velocity)
 
   return display_audio(wav, seconds)