Update inductor.py

inductor.py

@@ -96,7 +96,7 @@ class BartInductor(object):
                 ret = [t[0].replace('<ent0>','<mask>').replace('<ent1>','<mask>') for t in tB_probs]
                 new_ret = []
                 for temp in ret:
-                    temp = self.clean(temp[0].strip())
+                    temp = self.clean(temp.strip())
                     if len(new_ret) < topk and temp not in new_ret:
                         new_ret.append(temp)