Update app.py

app.py

@@ -28,6 +28,7 @@ def detect(opt, save_img=False):
 #     from utils.plots import plot_one_box
 #     from utils.torch_utils import select_device, load_classifier, time_synchronized, TracedModel
 
+    from yolov7_package import Yolov7Detector
     from yolov7_package.models.experimental import attempt_load
     from yolov7_package.utils.general import check_img_size, check_requirements, check_imshow, non_max_suppression, apply_classifier, \
         scale_coords, xyxy2xywh, strip_optimizer, set_logging, increment_path
@@ -50,7 +51,9 @@ def detect(opt, save_img=False):
     half = device.type != 'cpu'  # half precision only supported on CUDA
 
     # Load model
+    det = Yolov7Detector(weights=weights, traced=False)
     model = attempt_load(weights, map_location=device)  # load FP32 model
+    model = TracedModel(model, device, 640)
     stride = int(model.stride.max())  # model stride
     imgsz = check_img_size(imgsz, s=stride)  # check img_size