Update app.py

app.py

@@ -9,8 +9,8 @@ max_len = 10
 def ocr(plate_image, ocr_model):
   try:
     if 'yolo' in ocr_model:
-      # model = YOLO(f'./weights/{ocr_model}.pt', task='detect')
-      model = YOLO(f'./weights/best.pt', task='detect')
+      model = YOLO(f'./weights/{ocr_model}.pt', task='detect')
+      # model = YOLO(f'./weights/best.pt', task='detect')
       preds = model.predict(plate_image)
       pred_data = [list(pred.boxes.data.cpu().numpy()[0]) for pred in preds[0]]
       x_sorted_preds = sorted(pred_data, key=lambda x:x[0])