Spaces:

ejschwartz
/

function-method-detector

Sleeping

Edward J. Schwartz commited on Sep 15, 2023

Commit

356346e

1 Parent(s): 9953197

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,7 +131,7 @@ with gr.Blocks() as demo:
         out = model_wrap(text)
         print(out, file=sys.stderr)
         explainer = shap.Explainer(model_wrap)
-        shap_values = explainer([text])
         # Dimensions are (batch size, text size, number of classes)
         # Since we care about positive sentiment, use index 1

         out = model_wrap(text)
         print(out, file=sys.stderr)
         explainer = shap.Explainer(model_wrap)
+        shap_values = explainer(text)
         # Dimensions are (batch size, text size, number of classes)
         # Since we care about positive sentiment, use index 1