Update streamlit and clean up requirements.txt, pin pandas
Browse files- app.py +3 -3
- requirements.txt +2 -14
app.py
CHANGED
@@ -12,7 +12,7 @@ IMAGE_WIDTHS = 900
|
|
12 |
PRE_TRAINED_DB = "data/pretrained.sqlite"
|
13 |
|
14 |
|
15 |
-
@st.
|
16 |
def load_eval_data():
|
17 |
conn = sqlite3.connect(PRE_TRAINED_DB)
|
18 |
conn.row_factory = lambda c, r: {
|
@@ -343,7 +343,7 @@ mT5 green and the other models black.
|
|
343 |
"Choose right summarization model", df["name"], index=33
|
344 |
)
|
345 |
|
346 |
-
@st.
|
347 |
def get_row(model):
|
348 |
return df[df["name"] == model]
|
349 |
|
@@ -367,7 +367,7 @@ mT5 green and the other models black.
|
|
367 |
with col2:
|
368 |
trans_model_right = st.selectbox("Choose right model", df["name"], index=32)
|
369 |
|
370 |
-
@st.
|
371 |
def get_row(model):
|
372 |
return df[df["name"] == model]
|
373 |
|
|
|
12 |
PRE_TRAINED_DB = "data/pretrained.sqlite"
|
13 |
|
14 |
|
15 |
+
@st.cache_resource
|
16 |
def load_eval_data():
|
17 |
conn = sqlite3.connect(PRE_TRAINED_DB)
|
18 |
conn.row_factory = lambda c, r: {
|
|
|
343 |
"Choose right summarization model", df["name"], index=33
|
344 |
)
|
345 |
|
346 |
+
@st.cache_resource
|
347 |
def get_row(model):
|
348 |
return df[df["name"] == model]
|
349 |
|
|
|
367 |
with col2:
|
368 |
trans_model_right = st.selectbox("Choose right model", df["name"], index=32)
|
369 |
|
370 |
+
@st.cache_resource
|
371 |
def get_row(model):
|
372 |
return df[df["name"] == model]
|
373 |
|
requirements.txt
CHANGED
@@ -1,17 +1,5 @@
|
|
1 |
-
|
2 |
-
-f https://download.pytorch.org/whl/cu116
|
3 |
-
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
4 |
-
protobuf<3.20
|
5 |
-
pandas
|
6 |
-
torch
|
7 |
-
transformers>=4.13.0
|
8 |
-
langdetect
|
9 |
psutil
|
10 |
-
jax[cuda]==0.3.16
|
11 |
-
chex>=0.1.4
|
12 |
-
##jaxlib==0.1.67
|
13 |
-
flax>=0.5.3
|
14 |
-
sentencepiece
|
15 |
matplotlib
|
16 |
seaborn
|
17 |
-
streamlit
|
|
|
1 |
+
pandas<2.0.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
psutil
|
|
|
|
|
|
|
|
|
|
|
3 |
matplotlib
|
4 |
seaborn
|
5 |
+
streamlit
|