Update segment_image.py

segment_image.py

@@ -182,11 +182,11 @@ class SegmentImage:
 
     def get_region_preds(self, img):
         """Function for predicting text region coordinates."""
-        results = self.region_model.segment(source=img,     
-                                            device=self.device, 
-                                            conf=self.region_conf_threshold, 
-                                            half=bool(self.region_half_precision), 
-                                            iou=self.region_nms_iou)
+        results = self.region_model(source=img,     
+                                    device=self.device, 
+                                    conf=self.region_conf_threshold, 
+                                    half=bool(self.region_half_precision), 
+                                    iou=self.region_nms_iou)
         results = results[0].cpu()
         if results.masks:
             # Extracts detected region polygons
@@ -211,11 +211,11 @@ class SegmentImage:
 
     def get_line_preds(self, img):
         """Function for predicting text line coordinates."""
-        results = self.line_model.segment(source=img, 
-                                        device=self.device, 
-                                        conf=self.line_conf_threshold, 
-                                        half=bool(self.line_half_precision),
-                                        iou=self.line_nms_iou)
+        results = self.line_model(source=img, 
+                                  device=self.device, 
+                                  conf=self.line_conf_threshold, 
+                                  half=bool(self.line_half_precision),
+                                  iou=self.line_nms_iou)
         results = results[0].cpu()
         if results.masks:
             # Detected text line polygons