Update app.py

app.py

@@ -70,7 +70,8 @@ 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
+    return results, dicom_image, dicom_meta
 
 def draw_bounding_boxes(img, results):
     for (x1, y1, x2, y2), conf, cls in results:
@@ -81,9 +82,12 @@ def draw_bounding_boxes(img, results):
     return img
 
 def show_preds_image(filepath):
-    results, img0 = detect_objects(filepath)
+    #results, img0 = detect_objects(filepath)
+    results, img0, dicom_meta = detect_objects(filepath)
     img_with_boxes = draw_bounding_boxes(img0, results)
-    return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB)
+    
+    #return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB)
+    return cv2.cvtColor(img_with_boxes, cv2.COLOR_BGR2RGB), dicom_meta
 
 def read_and_preprocess_dicom(file_path: str):
     """