Update app.py

app.py

@@ -33,7 +33,7 @@ for i, url in enumerate(file_urls):
         )
 
 
-model = YOLO('best.onnx', task='detect')
+model = YOLO('best.pt', task='detect')
 path  = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg'], ['image_3.jpg']]
 
 # path = [['IMG_7612.JPG'], ['IMG_7678.JPG'], ['all_33.jpg'], ['all_80.jpg'], 
@@ -49,7 +49,7 @@ classes = ['alligator_cracking', 'longitudinal_cracking', 'potholes', 'ravelling
 
 def show_preds_image(image_path):
     image = cv2.imread(image_path)
-    outputs = model.predict(source=image_path, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=1024)
+    outputs = model.predict(source=image_path, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=640)
     results = outputs[0].cpu().numpy()
 
     re_boxes = results.boxes.data.tolist()
@@ -117,7 +117,7 @@ def show_preds_video(video_path):
         ret, frame = cap.read()
         if ret:
             frame_copy = frame.copy()
-            outputs = model.predict(source=frame, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=1024)
+            outputs = model.predict(source=frame, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=640)
             results = outputs[0].cpu().numpy()
             re_boxes = results.boxes.data.tolist()