Removed storing input image on Cuda

app.py

@@ -27,7 +27,7 @@ yolo_grad_cam = YoloGradCAM(model=model, target_layers=[model.layers[-2]], use_c
 @torch.inference_mode()
 def detect_objects(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.4, enable_grad_cam: bool = False, transparency: float = 0.5) -> List[np.ndarray]:
     transformed_image = modelConfig.transforms(image=image)["image"].unsqueeze(0)
-    transformed_image = transformed_image.cuda()
+    #transformed_image = transformed_image.cuda()
     output = model(transformed_image)
 
     bboxes = [[] for _ in range(1)]