Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -195,7 +195,7 @@ def interpretation_function(file_obj):
|
|
195 |
clean_text = text_preprocessing(text)
|
196 |
clean_text = tokenizer.decode(tokenizer(clean_text)["input_ids"][:512])[6:]
|
197 |
explainer = shap.Explainer(document_classifier)
|
198 |
-
shap_values = explainer([clean_text])
|
199 |
|
200 |
# Dimensions are (batch size, text size, number of classes)
|
201 |
# Since we care about positive sentiment, use index 1
|
|
|
195 |
clean_text = text_preprocessing(text)
|
196 |
clean_text = tokenizer.decode(tokenizer(clean_text)["input_ids"][:512])[6:]
|
197 |
explainer = shap.Explainer(document_classifier)
|
198 |
+
shap_values = explainer([clean_text[:-10]])
|
199 |
|
200 |
# Dimensions are (batch size, text size, number of classes)
|
201 |
# Since we care about positive sentiment, use index 1
|