aidan-o-brien commited on
Commit
7a5750a
1 Parent(s): 1f6bd67

remove caching for choosing model

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -16,9 +16,8 @@ def load_pipeline(model_cp, tokenizer_cp):
16
  return pipeline("question-answering", model=model_cp, tokenizer=tokenizer_cp)
17
 
18
 
19
- @st.cache
20
  def choose_model():
21
- with st.sidebar:
22
  st.write("# Model Selection")
23
  model_cp = st.selectbox('Select model for inference',
24
  ('deepset/roberta-base-squad2',
16
  return pipeline("question-answering", model=model_cp, tokenizer=tokenizer_cp)
17
 
18
 
 
19
  def choose_model():
20
+ with st.sidebar: # streamlit doesn't know how to has this by default?
21
  st.write("# Model Selection")
22
  model_cp = st.selectbox('Select model for inference',
23
  ('deepset/roberta-base-squad2',