ashhadahsan commited on
Commit
915e04a
·
1 Parent(s): bab921c

added cache

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,6 +6,7 @@ from simplet5 import SimpleT5
6
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
 
8
 
 
9
  def load_t5():
10
  model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
11
 
@@ -41,7 +42,7 @@ ps = st.empty()
41
  if st.button("Process"):
42
  if uploaded_file is not None:
43
 
44
- df = pd.read_excel(uploaded_file,engine='openpyxl')
45
  columns = df.columns.values.tolist()
46
  columns = [x.lower() for x in columns]
47
  df.columns = columns
 
6
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
 
8
 
9
+ @st.cache
10
  def load_t5():
11
  model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
12
 
 
42
  if st.button("Process"):
43
  if uploaded_file is not None:
44
 
45
+ df = pd.read_excel(uploaded_file, engine="openpyxl")
46
  columns = df.columns.values.tolist()
47
  columns = [x.lower() for x in columns]
48
  df.columns = columns