diff --git a/enhancer/models/__init__.py b/enhancer/models/__init__.py
index 0d21337..534a608 100644
--- a/enhancer/models/__init__.py
+++ b/enhancer/models/__init__.py
@@ -1,2 +1,3 @@
 from enhancer.models.demucs import Demucs
-from enhancer.models.waveunet import WaveUnet
\ No newline at end of file
+from enhancer.models.waveunet import WaveUnet
+from enhancer.models.model import Model
\ No newline at end of file