update app

app.py

@@ -7,7 +7,6 @@ from transformers import AutoModelForSequenceClassification,AutoTokenizer,pipeli
 from transformers_interpret import SequenceClassificationExplainer
 import streamlit.components.v1 as components  # Import Streamlit
 
-@st.cache
 def visualize(text):
 
     checkpoint = 'mlkorra/OGBV-gender-bert-hi-en'
@@ -17,8 +16,8 @@ def visualize(text):
 
     cls_explainer = SequenceClassificationExplainer(model,tokenizer)
     word_attributions = cls_explainer(masked_text)
-    cls_explainer.visualize('visualize.html')
-    components.html('visualize.html')
+    components.html(cls_explainer.visualize('visualize.html'))
+    #components.html('visualize.html')
 
 @st.cache
 def load_model(text):