diff --git a/infer.py b/infer.py
index 02e9497..0e51de5 100644
--- a/infer.py
+++ b/infer.py
@@ -1,4 +1,3 @@
-from dataloader import DataLoader, DataLoaderInfer
 from argparse import ArgumentParser
 from tensorflow import keras
 import numpy as np