surdan
/

LaBSE_ner_nerel

Token Classification

Inference Endpoints

Model card Files Files and versions Community

fix index

#4

by GrigoryT22 - opened Mar 17

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

Inference.ipynb +1 -1

Inference.ipynb CHANGED Viewed

@@ -76,7 +76,7 @@
     "            if (ner_result[i][\"entity_group\"] == prev_entity) &\\\n",
     "               (ner_result[i][\"start\"] == prev_end):\n",
     "                \n",
-    "                entities[i-1][2] = ner_result[i][\"end\"]\n",
     "                prev_entity = ner_result[i][\"entity_group\"]\n",
     "                prev_end = ner_result[i][\"end\"]\n",
     "            else:\n",

     "            if (ner_result[i][\"entity_group\"] == prev_entity) &\\\n",
     "               (ner_result[i][\"start\"] == prev_end):\n",
     "                \n",
+    "                entities[-1][2] = ner_result[i][\"end\"]\n",
     "                prev_entity = ner_result[i][\"entity_group\"]\n",
     "                prev_end = ner_result[i][\"end\"]\n",
     "            else:\n",