Update app.py

app.py

@@ -163,7 +163,7 @@ input_choice = st.checkbox("Check this if you want to enter your own example to
 if input_choice == False:
     record_to_explain = st.number_input("Enter the index of the document from the original dataset to interpret", value = 30)
     #te.fit(df[column_name][record_to_explain], text_clf.predict_proba)
-    fit_text_explainer(df[column_name][record_to_explain], text_clf.predict_proba)
+    te = fit_text_explainer(df[column_name][record_to_explain], text_clf.predict_proba)
     if task == "Classification":
         st.write("Ground truth label")
         st.write(df[labels_column_name][record_to_explain])
@@ -179,7 +179,7 @@ if input_choice == False:
 else:
     record_to_explain = st.text_area("Enter the example document to explain", value = text_example)
     #te.fit(record_to_explain, text_clf.predict_proba)
-    fit_text_explainer(record_to_explain, text_clf.predict_proba)
+    te = fit_text_explainer(record_to_explain, text_clf.predict_proba)
     if task == "Classification":
         st.write("Model prediction")
         model_prediction = text_clf.predict([record_to_explain])