Spaces:
Runtime error
Runtime error
Commit
·
7a5750a
1
Parent(s):
1f6bd67
remove caching for choosing model
Browse files
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',
|