Update app.py

app.py

@@ -29,7 +29,8 @@ def detect(image):
         if seg_box[5] in vehicle_classes:
             vehicle_box = seg_box[:4].to(torch.int32)
             vehicle_crop = image.crop(vehicle_box.tolist())
-            lp_result = lp_model(vehicle_crop, imgsz=(vehicle_crop.height, vehicle_crop.width), device="cpu")[0]
+            imgsz = (vehicle_crop.height, vehicle_crop.width) if vehicle_crop.width < 640 and vehicle_crop.height < 640 else (640, 640)
+            lp_result = lp_model(vehicle_crop, imgsz=imgsz, device="cpu")[0]
             lp_boxes = lp_result.boxes.data[:, :4]
             vehicle_offset = torch.cat((vehicle_box[:2], vehicle_box[:2]))
             for lp_box in lp_boxes: