diff --git a/predict.py b/predict.py
index 5190528..a76679e 100644
--- a/predict.py
+++ b/predict.py
@@ -1,3 +1,12 @@
+import pandas
+import os
+
+from keras.applications.densenet import layers
+
+from sklearn.model_selection import train_test_split
+import tensorflow
+
+
 reloaded = tensorflow.keras.models.load_model('test')
 
 x_to_test = pandas.read_csv('./X_test.csv')