Update app.py

app.py

@@ -25,7 +25,8 @@ def shap(txt, tok_level):
         y_pred = model(txt)
         y_offs = model(batch)
         shaps = (y_offs - y_pred).tolist()[0] # convert to list and make tuple to be returned
-    return [token, shap for token, shap in list(zip(tokens, shaps))]
+    labels = ["+" if s < 0 else "-" for s in shaps]
+    return list(zip(tokens, labels))
 
 demo = gr.Interface(shap,
                     [