Update app.py

app.py

@@ -4,8 +4,9 @@ import cv2
 import numpy as np
 from PIL import Image
 from torchvision.transforms import functional as F
-from yolov5.utils.general import non_max_suppression
-from yolov5.models.yolo import Model
+from ultralytics.yolo.utils.ops import non_max_suppression
+from ultralytics.yolo.engine.model import Model
+
 
 # Load YOLOv5 model
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')