Made a pull request

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import numpy as np
 # from scipy.special import softmax
 # import os
-from utils import run_sentiment_analysis, preprocess
+from utils import run_sentiment_analysis, preprocess, integrated_gradients
 from transformers import AutoTokenizer, AutoConfig,AutoModelForSequenceClassification
 import os
 import time
@@ -97,8 +97,14 @@ with my_expander:
             time.sleep(5)
             success_message.empty()
     interpret_button = col2.button(label='Interpret',type='secondary', use_container_width=True)
+    if interpret_button:
+            ig = integrated_gradients(text, model)
+            
+            ig = ig.sum(dim=-1).squeeze().detach().numpy()[0]
+            ig = (ig - ig.min()) / (ig.max() - ig.min())
+            # Display the Integrated Gradients as a bar chart
+            st.bar_chart(ig)
 
 
 # st.help()
     # create a date input to receive date
-