Fixing issue with model load for yolov7

track.py

@@ -40,6 +40,8 @@ from ultralytics.yolo.utils.torch_utils import select_device
 from ultralytics.yolo.utils.ops import Profile, non_max_suppression, scale_boxes, process_mask, process_mask_native
 from ultralytics.yolo.utils.plotting import Annotator, colors, save_one_box
 
+from models.experimental import attempt_load
+
 from trackers.multi_tracker_zoo import create_tracker
 
 
@@ -105,8 +107,13 @@ def run(
     # Load model
     device = select_device(device)
     is_seg = '-seg' in str(yolo_weights)
+
+    if model_link== 'yolov8m':
+        model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
+    else:
+        model = attempt_load(yolo_weights, map_location=device) 
    
-    model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
+    #model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
     stride, names, pt = model.stride, model.names, model.pt
     imgsz = check_imgsz(imgsz, stride=stride)  # check image size