Update app.py

app.py

@@ -70,7 +70,6 @@ def detect_objects(image_path):
     img, img0 = preprocess_image(dicom_image)
     pred = infer(model, img)
     results = postprocess(pred, dicom_image, img)
-    #return results, dicom_image
     return results, dicom_image, dicom_meta
 
 def draw_bounding_boxes(img, results, dicom_meta):
@@ -81,11 +80,9 @@ def draw_bounding_boxes(img, results, dicom_meta):
         xc = x1+(x2-x1)/2
         yc = y1+(y2-y1)/2
         conf = round(conf.detach().item(), 4)
-        #print("results in draw_bounding box:",[(x1, y1, x2, y2), conf, cls])
 
-        #dets.append([(x1, y1, x2, y2), conf, cls])
         dets.append([(xc, yc, zc), conf, cls])
-        cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), 2)
+        cv2.rectangle(img, (x1, y1), (x2, y2), (0, 0, 255), 2)
         cv2.putText(img, f'{model.names[int(cls)]} {conf:.2f}', (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (36, 255, 12), 2)
     return img, dets