yhavinga commited on
Commit
bd572c4
1 Parent(s): 6b8d575

Update streamlit and clean up requirements.txt, pin pandas

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. 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.cache
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.cache
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.cache
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
- #-f https://download.pytorch.org/whl/torch_stable.html
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.10.0
1
+ pandas<2.0.0
 
 
 
 
 
 
 
2
  psutil
 
 
 
 
 
3
  matplotlib
4
  seaborn
5
+ streamlit