debug

app.py

@@ -123,10 +123,14 @@ with gr.Blocks() as demo:
         }
     
     def interpretation_function(text):
+
+        def model_wrap(input):
+            return model.fn(input)['confidences']
+
         print(text, file=sys.stderr)
-        what = model.fn(text)
-        print(what, file=sys.stderr)
-        explainer = shap.Explainer(model.fn)
+        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)