Update generic_ner.py

generic_ner.py

@@ -238,11 +238,11 @@ class MultitaskTokenClassificationPipeline(Pipeline):
             confidence_scores[task] = F.softmax(logits, dim=-1).tolist()[0]
 
         entities = {}
-        for task, preds in predictions.items():
+        for task in predictions.keys():
             words_list, preds_list, confidence_list = realign(
                 text_sentence,
-                preds[0],
-                confidence_scores[task][0],
+                predictions[task],
+                confidence_scores[task],
                 self.tokenizer,
                 self.id2label[task],
             )