app update

app.py

@@ -2,15 +2,15 @@ import torch
 import torch.optim as optim
 import lightning.pytorch as pl
 from torchvision import transforms
-from custom_yolo.custom_library.utils import cells_to_bboxes, non_max_suppression
-from custom_yolo.custom_library import config
+from custom_library.utils import cells_to_bboxes, non_max_suppression
+from custom_library import config
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
-from custom_yolo.custom_library.lightning_model import YOLOv3Lightning
+from custom_library.lightning_model import YOLOv3Lightning
 import cv2
 import numpy as np
 from pytorch_grad_cam.utils.image import show_cam_on_image
-from custom_yolo.custom_library.gradio_utils import  draw_predictions, YoloCAM
+from custom_library.gradio_utils import  draw_predictions, YoloCAM
 import gradio as gr
 import os
 import albumentations as A