fix last entity

app.py

@@ -96,6 +96,10 @@ def render_sample(subset, pii_type):
             ents.append(Span(doc, start, end, sample['labels'][start][2:]))
             in_entity = False
 
+    # Add the last entity if we're still in one
+    if in_entity:
+        ents.append(Span(doc, start, end, sample['labels'][start][2:]))
+
     doc.ents = ents
     output = displacy.render(doc, style="ent", jupyter=False, options=options)
     return {'document': sample['document']}, output
@@ -133,5 +137,3 @@ with gr.Blocks() as demo:
 
 # launch app ---
 demo.launch()
-
-# removed competition set ---