diff --git a/ium_predict.py b/ium_predict.py
index 2c91ead..42bac6d 100644
--- a/ium_predict.py
+++ b/ium_predict.py
@@ -1,4 +1,4 @@
-from keras.models import Sequential, load_model
+from tensorflow.keras.models import Sequential
 import pandas as pd
 from sklearn.preprocessing import LabelEncoder
 from sklearn import metrics