update app

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import streamlit as st
 from transformers import AutoModelForSequenceClassification,AutoTokenizer,pipeline
 from transformers_interpret import SequenceClassificationExplainer
+import streamlit.components.v1 as components  # Import Streamlit
 
 @st.cache
 def visualize(text):
@@ -14,10 +15,10 @@ def visualize(text):
     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')
-     
+    cls_explainer = SequenceClassificationExplainer(model,tokenizer)
+    word_attributions = cls_explainer(masked_text)
+    cls_explainer.visualize('visualize.html')
+    components.html('visualize.html')
 
 @st.cache
 def load_model(text):