Spaces:

ejschwartz
/

function-method-detector

Sleeping

Edward J. Schwartz commited on Sep 15, 2023

Commit

db8ba99

1 Parent(s): 4c715cf

Try this

Files changed (1) hide show

app.py CHANGED Viewed

@@ -122,7 +122,7 @@ with gr.Blocks() as demo:
         }
     def interpretation_function(text):
-        explainer = shap.Explainer(model)
         shap_values = explainer([text])
         # Dimensions are (batch size, text size, number of classes)

         }
     def interpretation_function(text):
+        explainer = shap.Explainer(model.fn)
         shap_values = explainer([text])
         # Dimensions are (batch size, text size, number of classes)