fix reference to pred_dict

app.py

@@ -33,11 +33,11 @@ labels = learn.dls.vocab
 
 
 def get_content_area(pred_dict) -> int:
-    if "content" not in pred_dict["labels"]:
+    if "content" not in pred_dict["detection"]["labels"]:
         return 0
     content_bboxes = [
-        pred_dict["bboxes"][idx]
-        for idx, label in enumerate(pred_dict["labels"])
+        pred_dict["detection"]["bboxes"][idx]
+        for idx, label in enumerate(pred_dict["detection"]["labels"])
         if label == "content"
     ]
     cb = content_bboxes[0]
@@ -45,11 +45,11 @@ def get_content_area(pred_dict) -> int:
 
 
 def get_redaction_area(pred_dict) -> int:
-    if "redaction" not in pred_dict["labels"]:
+    if "redaction" not in pred_dict["detection"]["labels"]:
         return 0
     redaction_bboxes = [
-        pred_dict["bboxes"][idx]
-        for idx, label in enumerate(pred_dict["labels"])
+        pred_dict["detection"]["bboxes"][idx]
+        for idx, label in enumerate(pred_dict["detection"]["labels"])
         if label == "redaction"
     ]
     return sum(
@@ -137,7 +137,7 @@ def predict(pdf, confidence, generate_file):
                     font_size=16,
                     label_color="#FF59D6",
                 )
-                print(pred_dict)
+                # print(pred_dict)
 
                 total_image_areas += pred_dict["width"] * pred_dict["height"]
                 total_content_areas += get_content_area(pred_dict)