scd

utilitis.py

@@ -81,7 +81,10 @@ def get_word(bboxes,image):
     roi = image.crop((x_min, y_min, x_max, y_max)) # Region of intrest
     roi_np = np.array(roi) # To array
     result = ocr.ocr(roi_np, cls=True) # Apply OCR to ROI
-    return result[0][0][1][0]
+    if result != [None]:
+        return result[0][0][1][0]
+    else:
+        return ""
 #############################################################################
 #############################################################################  
 def get_Finale_results(offset_mapping,id2label,image,prediction_scores,predictions,token_boxes):