diff --git a/nodes.py b/nodes.py
index eb734b8..62c3cd1 100644
--- a/nodes.py
+++ b/nodes.py
@@ -3,7 +3,6 @@
 import cv2
 import numpy as np
 import torch
-import torchvision.transforms as transforms
 
 from .PixelOE.pixeloe import pixelize
 
@@ -84,15 +83,9 @@ def INPUT_TYPES(cls):
 
     def process(self, img, mode, target_size, patch_size, thickness, color_matching, contrast, saturation, colors, no_upscale, no_downscale):
 
-        # Convert image from PyTorch tensor to NumPy array
         img = img.squeeze().numpy()
-
-        # Convert image from float32 to uint8
         img = (img * 255).astype(np.uint8)
-
-        # Convert image from RGB to BGR
         img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-        print(img.shape)
 
         img_pix = pixelize(img, mode, target_size, patch_size, thickness, color_matching, contrast, saturation, colors, no_upscale, no_downscale)
 
diff --git a/requirements.txt b/requirements.txt
index bc12b27..c7b41e4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,3 @@
 torch
-torchvision
 opencv-python
 numpy