Update app.py

app.py

@@ -66,6 +66,7 @@ import tensorflow as tf
 from tensorflow.keras.preprocessing.sequence import pad_sequences
 import spacy
 from tensorflow.keras.preprocessing.text import Tokenizer
+import streamlit_ext as ste
 #from spacy import en_core_web_lg
 #import en_core_web_lg
 #nlp = en_core_web_lg.load()
@@ -560,7 +561,7 @@ def main():
 
   with st.container():
     
-    st.download_button(label="Download the result table",data=buffer,file_name="t2cg_outputs.xlsx",mime="application/vnd.ms-excel")
+    ste.download_button(label="Download the result table",data=buffer,file_name="t2cg_outputs.xlsx",mime="application/vnd.ms-excel")
     st.markdown('<a href="https://huggingface.co/spaces/KGBrain/visual-knowledgegraph" target="_blank">Click this link in a separate tab to view knowledge graph</a>', unsafe_allow_html=True)
     # st.download_button(label="Download the detailed result table_csv",data=csv1,file_name='results.csv',mime='text/csv')
     # st.download_button(label="Download the result table_csv",data=csv2,file_name='final_data.csv',mime='text/csv')