Update display.py

display.py

@@ -7,6 +7,7 @@ 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, non_max_suppression
+from model import YOLOv3
 
 def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75,show_cam: bool = False,  transparency: float = 0.5):
     model = YOLOv3(num_classes=20)