Xrenya commited on
Commit
00fac80
1 Parent(s): 267b4f9

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -193,7 +193,7 @@ def plot_weights(file_obj):
193
  def interpretation_function(file_obj):
194
  text = read_file(file_obj)
195
  clean_text = text_preprocessing(text)
196
- clean_text = tokenizer.decode(tokenizer(clean_text)["input_ids"][6:512])
197
  explainer = shap.Explainer(document_classifier)
198
  shap_values = explainer([clean_text[:-20]])
199
 
 
193
  def interpretation_function(file_obj):
194
  text = read_file(file_obj)
195
  clean_text = text_preprocessing(text)
196
+ clean_text = " ".join(tokenizer.decode(tokenizer(" ".join(clean_text))["input_ids"]).split()[1:500])
197
  explainer = shap.Explainer(document_classifier)
198
  shap_values = explainer([clean_text[:-20]])
199