Update app.py

app.py

@@ -78,13 +78,15 @@ def draw_bounding_boxes(img, results):
         x1, y1, x2, y2, cls = map(int, [x1, y1, x2, y2, cls])
         conf = round(conf.detach().item(), 4)
         print("results in draw_bounding box:",[(x1, y1, x2, y2), conf, cls])
+        results = [(x1, y1, x2, y2), conf, cls])
         cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), 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
+    return img, results
 
 def show_preds_image(filepath):
     results, img0, dicom_meta = detect_objects(filepath)
-    img_with_boxes = draw_bounding_boxes(img0, results)
+    img_with_boxes, results = draw_bounding_boxes(img0, results)
+    
     #return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB)
     return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB), results, dicom_meta