Spaces:
Running
Running
ashhadahsan
commited on
Commit
·
915e04a
1
Parent(s):
bab921c
added cache
Browse files
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=
|
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
|