Update app.py

app.py

@@ -5,7 +5,7 @@ import cv2
 # Yolov8 nano on custom data with BS=32
 # model = YOLO('custom_model.pt')
 # Use YoloV8 Medium model trained on Custom data with bs32
-model = YOLO('yolov8m_bs32.pt')
+model = YOLO('best.pt')
 
 def infer(path):
     img = cv2.imread(path)