Update inductor.py

inductor.py

@@ -85,11 +85,12 @@ class BartInductor(object):
     def generate(self, inputs, k=10, topk=10):
         with torch.no_grad():
             tB_probs = self.generate_rule(inputs, k)
-            ret = [t[0].replace('<ent0>','<mask>').replace('<ent1>','<mask>') for t in tB_probs]
+            #ret = [t[0].replace('<ent0>','<mask>').replace('<ent1>','<mask>') for t in tB_probs]
+            ret = [(t[0].replace('<ent0>','<mask>').replace('<ent1>','<mask>'), t[1]) for t in tB_probs]
 
             new_ret = []
             for temp in ret:
-                temp = self.clean(temp.strip())
+                temp = (self.clean(temp[0].strip()), temp[1])
                 if len(new_ret) < topk and temp not in new_ret:
                     new_ret.append(temp)