Update display.py

display.py

@@ -6,7 +6,7 @@ import random
 from albumentations.pytorch import ToTensorV2
 from yolov3 import YOLOV3_PL
 from pytorch_grad_cam.utils.image import show_cam_on_image
-from utils import YoloCAM, cells_to_bboxes, draw_predictions, non_max_suppression
+from utils import YoloCAM, cells_to_bboxes, non_max_suppression
 
 def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75,  transparency: float = 0.5):
     model = YOLOV3_PL()