bug fix

upos.py

@@ -21,7 +21,7 @@ class BellmanFordTokenClassificationPipeline(TokenClassificationPipeline):
     z=e/e.sum(axis=-1,keepdims=True)
     for i in range(m.shape[0]-1,0,-1):
       m[i-1]+=numpy.nanmax(m[i]+self.transition,axis=1)
-    k=[numpy.nanargmax(m[0])]
+    k=[numpy.nanargmax(m[0]+self.transition[0])]
     for i in range(1,m.shape[0]):
       k.append(numpy.nanargmax(m[i]+self.transition[k[-1]]))
     w=[{"entity":self.model.config.id2label[j],"start":s,"end":e,"score":z[i,j]} for i,((s,e),j) in enumerate(zip(model_outputs["offset_mapping"][0].tolist(),k)) if s<e]