update app

app.py

@@ -20,7 +20,7 @@ def load_model(text):
   
     results = nlp(text)
     
-    return results
+    return results,model,tokenizer
     #MASK_TOKEN = tokenizer.mask_token
     #masked_text = masked_text.replace("<mask>", MASK_TOKEN)
     #result_sentence = nlp(masked_text)
@@ -53,9 +53,15 @@ def app():
     # pd.set_option('max_colwidth',30)
     if st.button("Classify"):
         with st.spinner("Classifying the sentence..."):
-            pred = load_model(masked_text)
+            pred,model,tokenizer = load_model(masked_text)
             st.write(pred)
 
+            if st.button('Visualize attributions'):
+                with st.spinner("Visualizing .....") :
+
+                cls_explainer = SequenceClassificationExplainer(model,tokenizer)
+                word_attributions = cls_explainer(masked_text)
+                st.write(cls_explainer.visualize())
 
 if __name__ == "__main__":
     app()