Update app.py

app.py

@@ -131,6 +131,9 @@ def detect_Custom(img):
 
                     if save_img or view_img:
                         label = f'{names[int(cls)]} {conf:.2f}'
+                        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)
+                        W, H = 300, 300
+                        box_voc = pbx.convert_bbox(xyxy, from_type="yolo", to_type="voc", image_size=(W,H))
                         if(cls == 1):
                           x1 = int(xyxy[0].item())
                           y1 = int(xyxy[1].item())
@@ -140,9 +143,6 @@ def detect_Custom(img):
                           orig_img = im0
                           crop_img = im0[y1:y2, x1:x2]
                           cv2.imwrite('MRZ_1.png', crop_img)
-                        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)
-                        W, H = 300, 300
-                        box_voc = pbx.convert_bbox(xyxy, from_type="yolo", to_type="voc", image_size=(W,H))
                         
             if view_img:
                 cv2.imshow(str(p), im0)