diff --git a/decensor.py b/decensor.py
index 48f5c65..a33ee97 100644
--- a/decensor.py
+++ b/decensor.py
@@ -11,6 +11,7 @@ from model import Model
 from poisson_blend import blend
 from config import *
 
+#TODO: allow variable batch sizes when decensoring. changing BATCH_SIZE will likely result in crashing
 BATCH_SIZE = 1
 
 mask_color = [0, 255, 0]
@@ -81,5 +82,4 @@ def get_mask(x_batch):
 if __name__ == '__main__':
     if not os.path.exists(args.decensor_output_path):
         os.makedirs(args.decensor_output_path)
-    decensor(args)
-    
+    decensor(args)
\ No newline at end of file