Update app.py

app.py

@@ -40,9 +40,15 @@ def text_line_detection_function(img):
     print("Text Line Detection Function Called")
     preds = batch_detection([img], det_model, det_processor)[0]  # Assuming this returns a DetectionResult object
     print(f"Detection Predictions: {preds}")
-    # Adjusted access to properties of preds
-    img_with_lines = draw_polys_on_image(preds.polygons, img)  # Assuming preds has a .polygons attribute
-    return img_with_lines, preds
+    
+    # Check if preds has an attribute 'bboxes' and use it
+    if hasattr(preds, 'bboxes'):
+        # Assuming draw_polys_on_image can work with the format of bboxes directly or you adapt it accordingly
+        img_with_lines = draw_polys_on_image([bbox.polygon for bbox in preds.bboxes], img)
+        return img_with_lines, preds
+    else:
+        raise AttributeError("DetectionResult object does not have 'bboxes' attribute")
+
 
 
 with gr.Blocks() as app: