Update app.py

app.py

@@ -105,7 +105,7 @@ def detect_objects(image_path):
     pred = infer(model, img)
     #results = postprocess(pred, img0.shape, img)
     results = postprocess(pred, dicom_image, img)
-    return results
+    return results, dicom_image
 
 def draw_bounding_boxes(img, results):
     for (x1, y1, x2, y2), conf, cls in results:
@@ -115,8 +115,8 @@ def draw_bounding_boxes(img, results):
     return img
 
 def show_preds_image(filepath):
-    dicom_image, dicom_meta = read_and_preprocess_dicom(image_path)
-    results = detect_objects(filepath)
+    #dicom_image, dicom_meta = read_and_preprocess_dicom(image_path)
+    results, img0 = detect_objects(filepath)
     #img0 = cv2.imread(filepath)
     #img_with_boxes = draw_bounding_boxes(img0, results)
     img_with_boxes = draw_bounding_boxes(img0, results)