diff --git a/get_emo.py b/get_emo.py
index b0fd448..d095980 100644
--- a/get_emo.py
+++ b/get_emo.py
@@ -1,4 +1,4 @@
-from emo_gen import RegressionHead, EmotionModel, AudioDataset, process_func
+from emo_gen import EmotionModel, process_func
 
 import librosa
 import numpy as np