Update app.py

app.py

@@ -148,11 +148,11 @@ def process_image(image):
     draw = ImageDraw.Draw(image)
     font = ImageFont.load_default()
 
-    def iob_to_label(label):
-        label = label[2:]
-        if not label:
+    def iob_to_label(labels):
+        labels= labels[2:]
+        if not labels:
             return 'other'
-    return label
+    return labels
 
     label2color = {'question':'blue', 'answer':'green', 'header':'orange', 'other':'violet'}
     for prediction, box in zip(true_predictions, true_boxes):