Update Tagger.py

Tagger.py

@@ -77,7 +77,7 @@ def draw_line_above_sent(folderpath, sent, para_ind, color=(0,0,0)):
     line_data = eval(data_df[data_df['Pg Ind'] == pg_ind]['Lines'].tolist()[0])[line_ind]
     line_bbox = line_data[0:-1]
     image = cv2.imread(folderpath + '/' + str(pg_ind) + '-processed.png')
-    image = cv2.line(image, (line_bbox[0] - 10, line_bbox[1]), (line_bbox[2] + 10, line_bbox[1]), thickness=2)
+    image = cv2.line(image, (line_bbox[0] - 10, line_bbox[1]), (line_bbox[2] + 10, line_bbox[1]), color=color, thickness=2)
     cv2.imwrite(folderpath + '/' + str(pg_ind) + '-processed.png', image)
 
 def process_file(folderpath, draw=False):