diff --git a/main_pipeline.py b/main_pipeline.py
index e6c239b..3fac520 100644
--- a/main_pipeline.py
+++ b/main_pipeline.py
@@ -72,7 +72,9 @@ if __name__ == '__main__':
     test_dataset = ShapeNetPartSegDataset('data', mode=GlobalVar.data_split.predict, collate_per_segment=False,
                                           refresh=True, transform=transforms)  # , cluster_type='pc')
 
-    grid_clusters = cluster_cubes(test_dataset[0], [1, 1, 1], max_points_per_cluster=8192)
+
+    # Here you can choose which dataset to load, in case there are more than one pointcloud files in raw/predict
+    grid_clusters = cluster_cubes(test_dataset[0], [1, 1, 1], max_points_per_cluster=4096)
 
     ps.init()
 
diff --git a/requirements.txt b/requirements.txt
index b01caa8..a456de7 100644
Binary files a/requirements.txt and b/requirements.txt differ