Fixed import statement with the usage of paths

audio.py

@@ -2,7 +2,6 @@
 import tensorflow as tf
 import numpy as np
 import pandas as pd
-from model import VAE 
 
 # Audio
 import pretty_midi
@@ -10,6 +9,16 @@ import pretty_midi
 # Displaying
 from IPython import display
 
+# Get the absolute path of the directory and add it to sys.path in order to
+# get the VAE class type
+import sys
+from pathlib import Path
+
+directory = Path(__file__).resolve().parent
+sys.path.insert(0, str(directory))
+
+from model import VAE
+
 # Extras
 import collections