Update gradio_utils.py

gradio_utils.py

@@ -30,9 +30,8 @@ icons = [
 
 icons_mapping = {config.PASCAL_CLASSES[i]:icons[i] for i in range(len(icons))}
 
-YOLO = YOLOv3Lightning()
-model = YOLO.load_from_checkpoint('YoLoV3Model2.ckpt',
-                                   map_location=torch.device('cpu'))
+model = YOLOv3Lightning.load_from_checkpoint('YoLoV3Model2.ckpt',
+                                              map_location=torch.device('cpu'))
 model.eval()
 
 scaled_anchors = (