Update app.py

app.py

@@ -62,12 +62,6 @@ def process_final_text(text):
     scores = output[0][0].detach().numpy()
     scores = tf.nn.softmax(scores)
     
-    # Transformer Interpret:
-    word_attributions = cls_explainer(X_test)
-    intp = pd.DataFrame(word_attributions)
-    intp.columns = ["Token",'Score']
-    intp_dict = intp.set_index('Token').T.to_dict('list')
-    
     # Get Keywords:
     keywords = custom_kw_extractor.extract_keywords(X_test)
     letter = []
@@ -96,12 +90,19 @@ def process_final_text(text):
         + sp_html
         + ""
     )
+    
+    # Transformer Interpret:
+    word_attributions = cls_explainer(X_test)
+    intp = pd.DataFrame(word_attributions)
+    intp.columns = ["Token",'Score']
+    intp_dict = intp.set_index('Token').T.to_dict('list')
+    
     return {"Resilience": float(scores.numpy()[1]), "Non-Resilience": float(scores.numpy()[0])},keywords,NER,intp_dict
     
 def main(prob1):
     text = str(prob1)
     obj = process_final_text(text)
-    return obj[0],obj[1],obj[2]
+    return obj[0],obj[1],obj[2],obj[3]
     
 title = "Welcome to **ResText** 🪐"
 description1 = """