diff --git a/utils/export_rtdetr_pytorch.py b/utils/export_rtdetr_pytorch.py
index 140cc71..600dd44 100755
--- a/utils/export_rtdetr_pytorch.py
+++ b/utils/export_rtdetr_pytorch.py
@@ -20,17 +20,6 @@ class DeepStreamOutput(nn.Module):
         scores, classes = torch.max(x['pred_logits'], 2, keepdim=True)
         classes = classes.float()
         return boxes, scores, classes
-    
-
-class DeepStreamInput(nn.Module):
-    def __init__(self, img_size, device):
-        self.img_size = img_size
-        self.device = device
-        super().__init__()
-
-    def forward(self, x):
-        size = torch.tensor([[*self.img_size]]).to(self.device)
-        return x, size
 
 
 def suppress_warnings():