Upload app.py

app.py

@@ -193,9 +193,9 @@ def plot_weights(file_obj):
 def interpretation_function(file_obj):
     text = read_file(file_obj)
     clean_text = text_preprocessing(text)
-    clean_text = tokenizer.decode(tokenizer(clean_text)["input_ids"][:512])[6:]
+    clean_text = tokenizer.decode(tokenizer(clean_text)["input_ids"][6:512])
     explainer = shap.Explainer(document_classifier)
-    shap_values = explainer([clean_text[:-10]])
+    shap_values = explainer([clean_text[:-20]])
 
     # Dimensions are (batch size, text size, number of classes)
     # Since we care about positive sentiment, use index 1