Update app.py

app.py

@@ -135,10 +135,16 @@ def render_table(forms, lemmas, upos, xpos, feats, named_entities):
     feats = [feat + [""] * (max_len - len(feat)) for feat in feats]
     feats = list(zip(*feats))
 
-    named_entities = [
-        "" if ne == "O" else f"<< {ne.split('-')[1]} >>" if ne.startswith("B") else ne.split('-')[1] if (ne.startswith("I") and i + 1 < len(named_entities) and named_entities[i + 1].startswith("I")) else f"{ne.split('-')[1]} >>"
-        for i, ne in enumerate(named_entities)
-    ]
+    named_entities = []
+    for i, ne in enumerate(named_entities):
+        if ne == "O":
+            named_entities.append("")
+        if ne.startswith("B"):
+            named_entities.append(f"<< {ne.split('-')[1]}")
+        elif ne.startswith("I") and i + 1 < len(named_entities) and named_entities[i + 1].startswith("I"):
+            named_entities.append(ne.split('-')[1])
+        else:
+            named_entities.append(f"{ne.split('-')[1]} >>")
 
     array = [
         [""] + forms,
@@ -175,10 +181,10 @@ with gr.Blocks(theme='sudeepshouche/minimalist', css=custom_css) as demo:
             dataset = gr.Dataset(components=[gr.Textbox(visible=False)],
                 label="Input examples",
                 samples=[
-                    ["Thomassen er på vei til sin neste gjerning."],
-                    ["På toppen av dette kom de metodiske utfordringer."],
-                    ["Berntsen har påtatt seg en både viktig og vanskelig oppgave."],
-                    ["Ikke bare har det vært et problem, som han selv skriver i forordet, å bli klok på Borten."],
+                    "Thomassen er på vei til sin neste gjerning.",
+                    "På toppen av dette kom de metodiske utfordringer.",
+                    "Berntsen har påtatt seg en både viktig og vanskelig oppgave.",
+                    "Ikke bare har det vært et problem, som han selv skriver i forordet, å bli klok på Borten.",
                 ]
             )