Update app.py

app.py

@@ -186,11 +186,11 @@ with gr.Blocks() as demo:
       for i in range(3):
           batch_size, A, S, _, _ = test_img_out[i].shape # 1, anchors = 3, scaling = 13/26/52
           anchor = anchors[i]
-          boxes_scale_i = utils_org.cells_to_bboxes(test_img_out[i], anchor, S=S, is_preds=True)
+          boxes_scale_i = utils.cells_to_bboxes(test_img_out[i], anchor, S=S, is_preds=True)
           for idx, (box) in enumerate(boxes_scale_i):
               bboxes[idx] += box
       # nms
-      nms_boxes = utils_org.non_max_suppression(bboxes[0], iou_threshold=0.6, threshold=0.5, box_format="midpoint",)
+      nms_boxes = utils.non_max_suppression(bboxes[0], iou_threshold=0.6, threshold=0.5, box_format="midpoint",)
 
       # use gradio image annotations
       height, width = 416, 416