diff --git a/datasets.py b/datasets.py
index 9f3e457..7a3f785 100644
--- a/datasets.py
+++ b/datasets.py
@@ -24,6 +24,9 @@ class CustomDataset(Dataset):
         self.dir_path = dir_path
         self.bbox_df = bbox_df
 
+        embed()
+        quit()
+
         self.all_images = np.array(sorted(self.bbox_df['image']), dtype=str)
         self.image_paths = list(map(lambda x: Path(self.dir_path)/x, self.all_images))
         # embed()
diff --git a/inference.py b/inference.py
index 47bb68c..27428b1 100644
--- a/inference.py
+++ b/inference.py
@@ -63,7 +63,6 @@ if __name__ == '__main__':
     test_data = create_train_or_test_dataset(TRAIN_DIR, train=False)
     test_loader = create_valid_loader(test_data)
 
-
     infere_model(test_loader, model)
 
     # detection_threshold = 0.8