added another model - model selection in sidebar

app.py

@@ -12,11 +12,24 @@ def load_data(file):
     return df
 
 
-#@st.cache # tokenier cannot be cached
 def load_pipeline(model_cp, tokenizer_cp):
     return pipeline("question-answering", model=model_cp, tokenizer=tokenizer_cp)
 
 
+@st.cache
+def choose_model():
+    with st.sidebar:
+        st.write("# Model Selection")
+        model_cp = st.selectbox('Select model for inference',
+                            ('deepset/roberta-base-squad2',
+                            'aidan-o-brien/recipe-improver'))
+        # If not my model > model_cp = tokenizer_cp, else > albert tokenizer
+        if model_cp == "aidan-o-brien/recipe-improver":
+            return model_cp, "albert-base-v2"
+        else:
+            return model_cp, model_cp
+
+
 # Page config
 title = "Recipe Improver"
 icon = "🍣"
@@ -24,9 +37,8 @@ st.set_page_config(page_title=title, page_icon=icon)
 st.title(title)
 
 
-# Load tokenizer and model
-model_cp = "aidan-o-brien/recipe-improver"
-tokenizer_cp = "albert-base-v2"
+# Load model and tokenzier
+model_cp, tokenizer_cp = choose_model()
 question_answer = load_pipeline(model_cp, tokenizer_cp)
 st.write("Model and tokenizer successfully loaded.")
 
@@ -39,7 +51,6 @@ with st.expander("Upload csv file"):
 # If file is uploaded, run inference
 if uploaded_file is not None:
     df = load_data(uploaded_file)
-    st.write(df.head())
 
     # Run inference on first example
     first_example = df['review'][0]