tsantos commited on
Commit
6450a99
·
1 Parent(s): b44729f

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -1
pipeline.py CHANGED
@@ -659,7 +659,8 @@ class Pipeline(object):
659
  return predictions,embeddings_output
660
 
661
  def bert_interpretability(self, input_text:str):
662
- return self.lime_interpretability.lime_analysis(self.bert_model,input_text, self.bert_model.clean_data(input_text), class_names=self.bert_model.config['classes'])
 
663
 
664
 
665
  if __name__ == '__main__':
 
659
  return predictions,embeddings_output
660
 
661
  def bert_interpretability(self, input_text:str):
662
+ clean_data,_ = self.bert_model.clean_data(input_text)
663
+ return self.lime_interpretability.lime_analysis(self.bert_model,input_text, clean_data, class_names=self.bert_model.config['classes'])
664
 
665
 
666
  if __name__ == '__main__':