Update app.py

app.py

@@ -18,7 +18,7 @@ def shap(txt, tok_level):
     #tokens = nltk
     with torch.no_grad():
         y_pred = model(txt)
-    return y_pred
+    return txt
 
 demo = gr.Interface(fn=shap,
              inputs=[