update app

app.py

@@ -9,10 +9,15 @@ from transformers_interpret import SequenceClassificationExplainer
 @st.cache
 def visualize(text):
 
+    checkpoint = 'mlkorra/OGBV-gender-bert-hi-en'
+
+    tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+    model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
+
      cls_explainer = SequenceClassificationExplainer(model,tokenizer)
      word_attributions = cls_explainer(masked_text)
      cls_explainer.visualize('visualize.html')
-
+     
 
 @st.cache
 def load_model(text):