Update Main.py

Main.py

@@ -134,7 +134,7 @@ class Main:
         label_ids = np.argmax(logits, axis=2)
         
         entities = []
-        for label_id, offset in zip(label_ids[0], self.data['offset_mapping']):
+        for label_id, offset in zip(label_ids[0], data['offset_mapping']):
             curr_id = self.idx2tag[label_id]
             curr_start = offset[0]
             curr_end = offset[1]